Program Listing for File kernel.h
↰ Return to documentation for file (include/runtime/include/api/kernel.h
)
#ifndef MINDSPORE_LITE_INCLUDE_KERNEL_H
#define MINDSPORE_LITE_INCLUDE_KERNEL_H
#include <vector>
#include <string>
#include <utility>
#include <map>
#include "schema/model_generated.h"
#include "include/api/types.h"
#include "include/api/context.h"
#include "include/api/kernel_api.h"
namespace mindspore::kernel {
class MS_API Kernel : public IKernel<schema::Primitive> {
public:
Kernel() = default;
Kernel(const std::vector<mindspore::MSTensor> &inputs, const std::vector<mindspore::MSTensor> &outputs,
const schema::Primitive *primitive, const mindspore::Context *ctx)
: IKernel<schema::Primitive>(inputs, outputs, primitive, ctx) {
Initialize();
}
virtual ~Kernel() = default;
virtual schema::PrimitiveType type() const { return type_; }
virtual schema::QuantType quant_type() const { return quant_type_; }
protected:
void Initialize();
protected:
schema::PrimitiveType type_ = schema::PrimitiveType_NONE;
schema::QuantType quant_type_ = schema::QuantType_QUANT_NONE;
};
} // namespace mindspore::kernel
#endif // MINDSPORE_LITE_INCLUDE_KERNEL_H