Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/utils/
Dconvert_utils.h38 using TensorPtr = std::shared_ptr<Tensor>; variable
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dcpu_session.cc160std::map<tensor::TensorPtr, session::KernelWithIndex> *tensor_to_node) { in CreateOutputTensors()
210 const std::vector<tensor::TensorPtr> &, VectorRef *const) { in PostExecuteGraph() argument
259 void CPUSession::UpdateDynamicOutputShape(const std::map<tensor::TensorPtr, KernelWithIndex> &tenso… in UpdateDynamicOutputShape()
Dpynative_task_manager.h35 std::map<tensor::TensorPtr, KernelWithIndex> tensor_to_node) in RunOpContext()
Dascend_session.cc767 const std::vector<tensor::TensorPtr> &, VectorRef *const) { in PostExecuteGraph() argument
815 const std::map<tensor::TensorPtr, session::KernelWithIndex> &tensor_to_node) const { in BindAddressToTensor() argument
834const std::map<tensor::TensorPtr, session::KernelWithIndex> &tensor_to_node, in LaunchFunc()
894std::map<tensor::TensorPtr, session::KernelWithIndex> *tensor_to_node, in PrepareForOutputTensor()
1488const std::map<tensor::TensorPtr, session::KernelWithIndex> &tensor_to_node, in RunOpMemoryAllocNew()
2010const std::map<tensor::TensorPtr, session::KernelWithIndex> &tensor_to_node, in UpdateOutputTensors()
Dsession_basic.cc198std::map<tensor::TensorPtr, session::KernelWithIndex> *tensor_to_node) { in CreateNodeOutputTensor()
251std::map<tensor::TensorPtr, session::KernelWithIndex> *tensor_to_node, in CreateNodeOutputTensors()
1607std::map<tensor::TensorPtr, session::KernelWithIndex> *tensor_to_node) const { in UpdateOutputs()
1691std::map<tensor::TensorPtr, session::KernelWithIndex> *tensor_to_node) { in CreateOutputTensors()
1712const std::map<tensor::TensorPtr, session::KernelWithIndex> &tensor_to_node, in UpdateOutputTensors()
Dgpu_session.cc564const std::map<tensor::TensorPtr, session::KernelWithIndex> &tensor_to_node, in UpdateOutputTensors()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/
Dtbe_property_checker.cc26 using TensorPtr = mindspore::tensor::TensorPtr; typedef
/third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/
Dcpu_kernel_runtime.cc188 std::map<tensor::TensorPtr, session::KernelWithIndex> *tensor_to_node) { in CreatTensorForOutput()
249std::map<tensor::TensorPtr, session::KernelWithIndex> *tensor_to_node) { in CreatTensorForOutput()
281std::map<tensor::TensorPtr, session::KernelWithIndex> *tensor_to_node) { in CreateOutputTensors()
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_runtime.h42 using TensorPtr = std::shared_ptr<Tensor>; variable
Dkernel_runtime.cc317const std::map<tensor::TensorPtr, session::KernelWithIndex> &tensor_to_node) { in RunOpAssignMemory()
437 …const AnfNodePtr &kernel, const std::map<tensor::TensorPtr, session::KernelWithIndex> &tensor_to_n… in RunOpAssignOutputMemory()
/third_party/mindspore/tests/ut/cpp/pynative/
Dpynative_execute_test.cc29 using TensorPtr = mindspore::tensor::TensorPtr; typedef
/third_party/mindspore/mindspore/lite/mindir/include/
Dmindir_lite_graph.h12 typedef void *TensorPtr; typedef
/third_party/mindspore/mindspore/core/ir/
Dtensor.h393 using TensorPtr = std::shared_ptr<Tensor>; variable
/third_party/mindspore/mindspore/lite/src/
Dtensor.h311 using TensorPtr = std::shared_ptr<mindspore::lite::Tensor>; variable
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/core/
Dtensor.h63 using TensorPtr = std::shared_ptr<Tensor>; variable
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/parse/
Ddata_converter.cc37 using TensorPtr = mindspore::tensor::TensorPtr; typedef
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dgraph_kernel_helper.cc208 …[&tensor](const std::pair<tensor::TensorPtr, AnfNodePtrList> &vl) { return vl.first->ValueEqual(*t… in ConvertNonscalarTensorToParameter()