Home
last modified time | relevance | path

Searched defs:CreateOutputTensors (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/hardware/
Dcpu_session.cc160 void CPUSession::CreateOutputTensors(const GraphId &graph_id, const std::vector<tensor::TensorPtr> … in CreateOutputTensors() function in mindspore::session::CPUSession
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/device/
Dcpu_kernel_runtime.cc282 void CPUKernelRuntime::CreateOutputTensors(session::KernelGraph *kernel_graph, in CreateOutputTensors() function in mindspore::device::cpu::CPUKernelRuntime
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dsparse_cross_cpu_kernel.cc305 void CreateOutputTensors(const std::vector<std::unique_ptr<ColumnInterface<T>>> &columns, int64_t b… in CreateOutputTensors() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dsparse_cross.cc348 void CreateOutputTensors(const std::vector<std::unique_ptr<ColumnInterface<InternalType>>> &columns… in CreateOutputTensors() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/
Dsession_basic.cc1032 void SessionBasic::CreateOutputTensors(const GraphId &graph_id, const std::vector<tensor::TensorPtr… in CreateOutputTensors() function in mindspore::session::SessionBasic