UniformRequantize.Options

public static class UniformRequantize.Options

Optional attributes for UniformRequantize

Public Methods

UniformRequantize.Options
inputQuantizationAxis(Long inputQuantizationAxis)
UniformRequantize.Options
outputQuantizationAxis(Long outputQuantizationAxis)

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 UniformRequantize.Options inputQuantizationAxis (Long inputQuantizationAxis)

Parameters
inputQuantizationAxis The quantization axis that was used when quantizing original data that `input` represents. 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. Otherwise, it must be set within range [0, input.dims()).

public UniformRequantize.Options outputQuantizationAxis (Long outputQuantizationAxis)

Parameters
outputQuantizationAxis The new quantization axis to use to quantize original data that `input` represents.