UniformQuantizedDot.Options

public static class UniformQuantizedDot.Options

Optional attributes for UniformQuantizedDot

Public Methods

UniformQuantizedDot.Options
lhsQuantizationAxis(Long lhsQuantizationAxis)
UniformQuantizedDot.Options
outputQuantizationAxis(Long outputQuantizationAxis)
UniformQuantizedDot.Options
rhsQuantizationAxis(Long rhsQuantizationAxis)

Inherited Methods

boolean
equals(Object arg0)
final Class<?>
getClass()
int
hashCode()
final void
notify()
final void
notifyAll()
String
toString()
final void
wait(long arg0, int arg1)
final void
wait(long arg0)
final void
wait()

Public Methods

public UniformQuantizedDot.Options lhsQuantizationAxis (Long lhsQuantizationAxis)

Parameters
lhsQuantizationAxis 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 dot op lhs, only per-tensor quantization is supported. Thus, this attribute must be set to -1. Other values are rejected.

public UniformQuantizedDot.Options outputQuantizationAxis (Long outputQuantizationAxis)

Parameters
outputQuantizationAxis 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 dot op output, only per-tensor quantization or per-channel quantization along dimension 1 is supported. Thus, this attribute must be set to -1 or 1. Other values are rejected.

public UniformQuantizedDot.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 dot op rhs, only per-tensor quantization or per-channel quantization along dimension 1 is supported. Thus, this attribute must be set to -1 or 1. Other values are rejected.