|
|
|
@@ -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_; |
|
|
|
|