|
|
|
@ -66,6 +66,8 @@ class LiteSession : public session::LiteSession {
|
|
|
|
|
int Resize(const std::vector<mindspore::tensor::MSTensor *> &inputs,
|
|
|
|
|
const std::vector<std::vector<int>> &dims) override;
|
|
|
|
|
|
|
|
|
|
void set_model(Model *model) { this->model_ = model; }
|
|
|
|
|
|
|
|
|
|
protected:
|
|
|
|
|
static void ConvertTensorsQuantParam(const schema::Tensor *src_tensor, lite::Tensor *dst_tensor);
|
|
|
|
|
|
|
|
|
@ -123,6 +125,7 @@ class LiteSession : public session::LiteSession {
|
|
|
|
|
// graph output tensor name -- output tensor
|
|
|
|
|
std::unordered_map<std::string, mindspore::tensor::MSTensor *> output_tensor_map_;
|
|
|
|
|
Executor *executor_ = nullptr;
|
|
|
|
|
Model *model_ = nullptr;
|
|
|
|
|
std::atomic<bool> is_running_ = false;
|
|
|
|
|
#if SUPPORT_GPU && !SUPPORT_TRAIN
|
|
|
|
|
opencl::OpenCLRuntimeWrapper ocl_runtime_wrap_;
|
|
|
|
|