Home
last modified time | relevance | path

Searched defs:TensorPtr (Results 1 – 20 of 20) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/mindapi/ir/
Dtensor.h91 using TensorPtr = SharedPtr<Tensor>; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/hardware/
Dcpu_session.cc162std::map<tensor::TensorPtr, session::KernelWithIndex> *tensor_to_node, in CreateOutputTensors()
208 const std::vector<tensor::TensorPtr> &, VectorRef *const) { in PostExecuteGraph() argument
234 void CPUSession::UpdateDynamicOutputShape(const std::map<tensor::TensorPtr, KernelWithIndex> &tenso… in UpdateDynamicOutputShape() argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/debug/data_dump/
Dstatistic_kernel.h41 using TensorPtr = tensor::TensorPtr; variable
Dtensor_statistic.h36 using TensorPtr = tensor::TensorPtr; variable
Dtensor_statistic.cc31 using TensorPtr = tensor::TensorPtr; typedef
Dstatistic_kernel.cc27 using TensorPtr = tensor::TensorPtr; typedef
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/device/
Dkernel_runtime.h42 using TensorPtr = std::shared_ptr<Tensor>; variable
Dkernel_runtime.cc329const std::map<tensor::TensorPtr, session::KernelWithIndex> &tensor_to_node) { in RunOpAssignMemory()
454const std::map<tensor::TensorPtr, session::KernelWithIndex> &tensor_to_node, in RunOpAssignOutputMemory()
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/pynative/
Dpynative_execute_test.cc34 using TensorPtr = mindspore::tensor::TensorPtr; typedef
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/c_api/src/
Dcommon.h70 using TensorPtr = mindspore::tensor::TensorPtr; variable
/third_party/mindspore/mindspore-src/source/mindspore/lite/mindir/include/
Dmindir_lite_graph.h12 typedef void *TensorPtr; typedef
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/device/
Dcpu_kernel_runtime.cc245 std::map<tensor::TensorPtr, session::KernelWithIndex> *tensor_to_node, in GetOrCreateTensorForOutput()
284std::map<tensor::TensorPtr, session::KernelWithIndex> *tensor_to_node) { in CreateOutputTensors()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/
Dsession_basic.cc129std::map<tensor::TensorPtr, session::KernelWithIndex> *tensor_to_node) { in CreateNodeOutputTensor()
381std::map<tensor::TensorPtr, session::KernelWithIndex> *tensor_to_node, in CreateNodeOutputTensors()
991std::map<tensor::TensorPtr, session::KernelWithIndex> *tensor_to_node) const { in UpdateOutputs()
1034std::map<tensor::TensorPtr, session::KernelWithIndex> *tensor_to_node, in CreateOutputTensors()
1050const std::map<tensor::TensorPtr, session::KernelWithIndex> &tensor_to_node, in UpdateOutputTensors()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/acl_ir/
Dge_adapter_info.h35 using TensorPtr = mindspore::tensor::TensorPtr; variable
/third_party/mindspore/mindspore-src/source/mindspore/core/ir/
Dtensor.h72 using TensorPtr = std::shared_ptr<Tensor>; variable
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/
Dtensor.h325 using TensorPtr = std::shared_ptr<mindspore::lite::Tensor>; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/core/
Dtensor.h58 using TensorPtr = std::shared_ptr<Tensor>; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/hardware/
Dgpu_session.cc545const std::map<tensor::TensorPtr, session::KernelWithIndex> &tensor_to_node, in UpdateOutputTensors()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/
Dconvert_utils_py.cc707 using TensorPtr = tensor::TensorPtr; typedef
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/parse/
Ddata_converter.cc49 using TensorPtr = mindspore::tensor::TensorPtr; typedef