public static class
UniformQuantizedConvolutionHybrid.Options
Optional attributes for UniformQuantizedConvolutionHybrid
Public Methods
| UniformQuantizedConvolutionHybrid.Options |
batchGroupCount(Long batchGroupCount)
|
| UniformQuantizedConvolutionHybrid.Options |
dimensionNumbers(String dimensionNumbers)
|
| UniformQuantizedConvolutionHybrid.Options |
explicitPadding(List<Long> explicitPadding)
|
| UniformQuantizedConvolutionHybrid.Options |
featureGroupCount(Long featureGroupCount)
|
| UniformQuantizedConvolutionHybrid.Options |
lhsDilation(List<Long> lhsDilation)
|
| UniformQuantizedConvolutionHybrid.Options |
rhsDilation(List<Long> rhsDilation)
|
| UniformQuantizedConvolutionHybrid.Options |
rhsQuantizationAxis(Long rhsQuantizationAxis)
|
| UniformQuantizedConvolutionHybrid.Options |
windowStrides(List<Long> windowStrides)
|
Inherited Methods
Public Methods
public UniformQuantizedConvolutionHybrid.Options batchGroupCount (Long batchGroupCount)
Parameters
| batchGroupCount | The number of batch groups. Used for grouped filters. Must be a divisor of output_feature. |
|---|
public UniformQuantizedConvolutionHybrid.Options dimensionNumbers (String dimensionNumbers)
Parameters
| dimensionNumbers | Structure of dimension information for the convolution op. Must be an empty string (default) or a serialized string of tensorflow.UniformQuantizedConvolutionDimensionNumbersAttr proto. If empty string, the default is `("NCHW", "OIHW", "NCHW")` (for a 2D convolution). |
|---|
public UniformQuantizedConvolutionHybrid.Options explicitPadding (List<Long> explicitPadding)
Parameters
| explicitPadding | If `padding` Attr is `"EXPLICIT"`, must be set as a list indicating
the explicit paddings at the start and end of each lhs spatial dimension.
Otherwise, this Attr is must be empty.
(If used,) Must be a list of size 2 * (number of lhs spatial dimensions), where (explicit_padding[2 * i], explicit_padding[2 * i + 1]) indicates spatial_dimensions[i] (start_padding, end_padding). |
|---|
public UniformQuantizedConvolutionHybrid.Options featureGroupCount (Long featureGroupCount)
Parameters
| featureGroupCount | The number of feature groups. Used for grouped convolutions. Must be a divisor of both lhs_feature and output_feature. |
|---|
public UniformQuantizedConvolutionHybrid.Options lhsDilation (List<Long> lhsDilation)
Parameters
| lhsDilation | The dilation factor to apply in each spatial dimension of `lhs`. Must be an empty list (default) or a list of size (number of lhs spatial dimensions). If empty list, the dilation for each lhs spatial dimension is set to 1. |
|---|
public UniformQuantizedConvolutionHybrid.Options rhsDilation (List<Long> rhsDilation)
Parameters
| rhsDilation | The dilation factor to apply in each spatial dimension of `rhs`. Must be an empty list (default) or a list of size (number of rhs spatial dimensions). If empty list, the dilation for each rhs spatial dimension is set to 1. |
|---|
public UniformQuantizedConvolutionHybrid.Options rhsQuantizationAxis (Long rhsQuantizationAxis)
Parameters
| rhsQuantizationAxis | Indicates the dimension index of the tensor where per-axis quantization is applied for the slices along that dimension. If set to -1 (default), this indicates per-tensor quantization. For the `rhs`, only per-tensor quantization or per-channel quantization along kernel_output_feature_dimension is supported. Thus, this attribute must be set to -1 or `dimension_numbers.kernel_output_feature_dimension`. Other values will raise error at OpKernel construction. |
|---|
public UniformQuantizedConvolutionHybrid.Options windowStrides (List<Long> windowStrides)
Parameters
| windowStrides | The stride of the sliding window for each spatial dimension of `lhs`. Must be an empty list (default) or a list of size (number of spatial dimensions). If an empty list is provided, the stride for each spatial dimension is set to 1. |
|---|