UniformQuantizedConvolutionHybrid.Options

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.