Home
last modified time | relevance | path

Searched defs:set_in_tensor (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/kernel/
Dbase_kernel.cc38 void BaseKernel::set_in_tensor(lite::Tensor *in_tensor, size_t index) { in set_in_tensor() function in mindspore::kernel::BaseKernel
Dextendrt_kernel_exec.h52 void set_in_tensor(lite::Tensor *in_tensor, size_t index) override { in set_in_tensor() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/
Dmatmul_fp32.h50 void set_in_tensor(lite::Tensor *in_tensor, size_t index) override { in set_in_tensor() function
Dconvolution_delegate_fp32.h62 void set_in_tensor(lite::Tensor *in_tensor, size_t index) override { in set_in_tensor() function
Dconvolution_sw_1x1_fp32.h57 void set_in_tensor(lite::Tensor *in_tensor, size_t index) override { in set_in_tensor() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/kernel/nnacl/
Dnnacl_base_kernel.h46 …void set_in_tensor(InferTensor *in_tensor, size_t index) override { lite_kernel_->set_in_tensor(in… in set_in_tensor() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp16/
Dconvolution_delegate_fp16.h71 void set_in_tensor(lite::Tensor *in_tensor, size_t index) override { in set_in_tensor() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/mindir_loader/mindir_model/
Dinner_kernel.h73 void set_in_tensor(lite::Tensor *in_tensor, size_t index) override { in set_in_tensor() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/
Dlite_kernel.h129 void set_in_tensor(lite::Tensor *in_tensor, size_t index) override { in set_in_tensor() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/executor/
Dkernel_exec.h228 virtual void set_in_tensor(lite::Tensor *in_tensor, size_t index) { in set_in_tensor() function