Class QuantDTypeCast

• Defined in File quant_dtype_cast.h

Inheritance Relationships

Base Type

• public mindspore::ops::BaseOperator

Class Documentation

class QuantDTypeCast : public mindspore::ops::BaseOperator

QuantDTypeCast QuantDTypeCast the QuantDTypeCast operator prototype.

Public Functions

inline QuantDTypeCast()

Constructor.

void Init(const int64_t src_t, const int64_t dst_t)

Method to init the op’s attributes.

参数

• src_t – [in] Define the data type of input.

• dst_t – [in] Define the data type of output.

void set_src_t(const int64_t src_t)

Method to set src_t attribute.

参数

src_t – [in] Define the data type of input.

int64_t get_src_t() const

Method to get src_t attribute.

返回

the data type of input.

void set_dst_t(const int64_t dst_t)

Method to set dst_t attribute.

参数

dst_t – [in] Define the data type of output.

int64_t get_dst_t() const

Method to get dst_t attribute.

返回

the data type of output.

void set_axis(const int64_t axis)

Method to set axis attribute.

参数

axis – [in] Define per-channel prefer dim.

int64_t get_axis() const

Method to get axis attribute.

返回

the per-channel prefer dim.