Class Kernel

• Defined in File kernel.h

Inheritance Relationships

Base Type

• public mindspore::kernel::IKernel< schema::Primitive > (Template Class IKernel)

Class Documentation

class Kernel : public mindspore::kernel::IKernel<schema::Primitive>

Public Functions

Kernel() = default

Constructor of Kernel.

inline Kernel(const std::vector<mindspore::MSTensor> &inputs, const std::vector<mindspore::MSTensor> &outputs, const schema::Primitive *primitive, const mindspore::Context *ctx)

Constructor of Kernel.

参数

• inputs – [in] Input tensors.

• outputs – [in] Output tensors.

• primitive – [in] The primitive of kernel.

• context – [in] The context created.

virtual ~Kernel() = default

Destructor of Kernel.

int InferShape() override

infer kernel's shape.

返回

status.

inline virtual schema::PrimitiveType type() const

obtain kernel's type.

返回

kernel's type.

inline virtual schema::QuantType quant_type() const

Obtain kernel's quant type.

返回

Kernel's quant type.