Home
last modified time | relevance | path

Searched defs:input_tensor_ (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/opencl/kernel/
Dpooling2d.h48 GpuTensorInfo input_tensor_; variable
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/string/
Dnormalize.cc38 Tensor input_tensor_; member in mindspore::TestNormalize
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/dataset/common/
Dcvop_common.h74 std::shared_ptr<Tensor> input_tensor_; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/
Dbatch_mat_mul.cc58 BaseTensorPtr input_tensor_ = input_tensor; in BatchMatMulAscendCustomize() local
Dmatmul.cc58 BaseTensorPtr input_tensor_ = input_tensor; in MatMulAscendCustomize() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/forward/
Dforward_task.h143 tensor::TensorPtr input_tensor_; variable