`neural_compressor.adaptor.ox_utils.operators.argmax`

ArgMax operator.

Module Contents

Classes

`ArgMaxOperator`	ArgMax operator.
`QArgMaxOperator`	INT8 ArgMax operator.

class neural_compressor.adaptor.ox_utils.operators.argmax.ArgMaxOperator(onnx_quantizer, onnx_node)[source]: ArgMax operator.

class neural_compressor.adaptor.ox_utils.operators.argmax.QArgMaxOperator(onnx_node, children, initializers)[source]: INT8 ArgMax operator.