Class FakeQuantWithMinMaxVarsPerChannel

Defined in File fake_quant_with_min_max_vars_per_channel.h

Inheritance Relationships

Base Type

public mindspore::ops::BaseOperator

Class Documentation

class FakeQuantWithMinMaxVarsPerChannel : public mindspore::ops::BaseOperator

Fake-quantize the input and one of shape: [d], [b, d], [b, h, w, d] by per-channel minimum and maximum. Refer to Python API mindspore.ops.FakeQuantWithMinMaxVarsPerChannel for more details.

Public Functions

inline FakeQuantWithMinMaxVarsPerChannel(): Constructor.

void Init(const int64_t num_bits = 8, const bool narrow_range = false): Init. Refer to the parameters of Python API mindspore.ops.FakeQuantWithMinMaxVarsPerChannel for the inputs.

void set_num_bits(const int64_t num_bits): Set num_bits.

void set_narrow_range(const bool narrow_range): Set narrow_range.

int64_t get_num_bits() const

Get num_bits.

Returns: num_bits.

bool get_narrow_range() const

Get narrow_range.

Returns: narrow_range.