Home
last modified time | relevance | path

Searched defs:input_tensors (Results 1 – 25 of 29) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/backend/session/
Dcpu_session.cc158 …sion::CreateOutputTensors(const GraphId &graph_id, const std::vector<tensor::TensorPtr> &input_ten… in CreateOutputTensors()
224 const std::vector<tensor::TensorPtr> &input_tensors, in BuildOpImpl()
274 std::vector<tensor::TensorPtr> *input_tensors, VectorRef *outputs, in RunOpImplOrigin()
280 std::vector<tensor::TensorPtr> *input_tensors, VectorRef *outputs, in RunOpImpl()
Dsession_basic.h75 std::vector<tensor::TensorPtr> input_tensors; member
231 const std::vector<tensor::TensorPtr> &input_tensors, in BuildOpImpl()
236 std::vector<tensor::TensorPtr> *input_tensors, VectorRef *outputs, in RunOpImpl()
239 std::vector<tensor::TensorPtr> *input_tensors, VectorRef *outputs, in RunOpImplOrigin()
Dpynative_task_manager.h34 std::vector<tensor::TensorPtr> input_tensors, in RunOpContext()
47 const std::vector<tensor::TensorPtr> &input_tensors() const { return input_tensors_; } in input_tensors() function
Dgpu_session.cc254 void GPUSession::RunOpAllocateMemory(const std::vector<tensor::TensorPtr> &input_tensors, in RunOpAllocateMemory()
639 const std::vector<tensor::TensorPtr> &input_tensors, in BuildOpImpl()
664 std::vector<tensor::TensorPtr> *input_tensors, VectorRef *outputs, in RunOpImplOrigin()
670 std::vector<tensor::TensorPtr> *input_tensors, VectorRef *outputs, in RunOpImpl()
Dascend_session.cc793 const std::vector<tensor::TensorPtr> &input_tensors, in BuildOpImpl()
835 … bool is_dynamic_shape, const std::vector<tensor::TensorPtr> &input_tensors) { in LaunchFunc()
893 const std::vector<tensor::TensorPtr> &input_tensors, in PrepareForOutputTensor()
915 std::vector<tensor::TensorPtr> *input_tensors, in CreateKernelGraph()
943 std::vector<tensor::TensorPtr> *input_tensors, VectorRef *outputs, in RunOpImpl()
986std::vector<tensor::TensorPtr> *input_tensors, VectorRef *outputs, in RunOpImplOrigin()
1024 const std::vector<tensor::TensorPtr> &input_tensors, in PreBuildOp()
1479 void AscendSession::RunOpMemoryAlloc(const std::vector<tensor::TensorPtr> &input_tensors, in RunOpMemoryAlloc()
1487 void AscendSession::RunOpMemoryAllocNew(const std::vector<tensor::TensorPtr> &input_tensors, in RunOpMemoryAllocNew()
Dsession_basic.cc158 const std::vector<tensor::TensorPtr> &input_tensors) { in GetNodeOutputTensorFromInputs()
197 const std::vector<tensor::TensorPtr> &input_tensors, in CreateNodeOutputTensor()
250 const std::vector<tensor::TensorPtr> &input_tensors, in CreateNodeOutputTensors()
377 const std::vector<tensor::TensorPtr> &input_tensors, in CreateNodeOutputPlaceholder()
409 const std::vector<tensor::TensorPtr> &input_tensors, in CreateNodeOutputPlaceholder()
1182 const std::vector<tensor::TensorPtr> &input_tensors) { in GetSingleOpGraphInfo()
1274 …const KernelGraphPtr &kernel_graph, const std::vector<tensor::TensorPtr> &input_tensors, VectorRef… in CreateOutputPlaceholder()
1606 const std::vector<tensor::TensorPtr> &input_tensors, in UpdateOutputs()
1689 …asic::CreateOutputTensors(const GraphId &graph_id, const std::vector<tensor::TensorPtr> &input_ten… in CreateOutputTensors()
2126const std::vector<tensor::TensorPtr> &input_tensors, in ConstructSingleOpGraph()
[all …]
Dkernel_graph.h280 …void SetInputTensors(const std::vector<tensor::TensorPtr> &input_tensors) { input_tensors_ = input… in SetInputTensors()
281 const std::vector<tensor::TensorPtr> &input_tensors() const { return input_tensors_; } in input_tensors() function
/third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/
Dloop_count_actor.cc87 std::vector<std::vector<TensorPtr>> input_tensors; in SendOutput() local
Ddata_prepare_actor.cc124 void DataPrepareActor::PrepareData(const std::vector<std::vector<TensorPtr>> &input_tensors, in PrepareData()
193 …taPrepareActor::PrepareDataForDeviceTensorStore(const std::vector<std::vector<TensorPtr>> &input_t… in PrepareDataForDeviceTensorStore()
224 …DataPrepareActor::PrepareDataForHostTensorQueue(const std::vector<std::vector<TensorPtr>> &input_t… in PrepareDataForHostTensorQueue()
268 void DataPrepareActor::PrepareDataForStepMode(const std::vector<std::vector<TensorPtr>> &input_tens… in PrepareDataForStepMode()
Dkernel_actor.cc134 const std::vector<TensorPtr> *input_tensors) { in RunOpControlWithInputTensor()
246 void KernelActor::PushInputDeviceTensor(const std::vector<TensorPtr> *input_tensors) { in PushInputDeviceTensor()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/
Dop_coder.cc30 const std::vector<Tensor *> OperatorCoder::input_tensors() const { return input_tensors_; } in input_tensors() function in mindspore::lite::micro::OperatorCoder
/third_party/mindspore/mindspore/lite/test/ut/src/
Dutils_test.cc64 auto input_tensors = kernel::LiteKernelUtil::SubgraphInputTensors(kernels); in TEST_F() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Daffine_fp32.cc258 std::vector<lite::Tensor *> input_tensors; in FullMatmulKernelCreate() local
312 std::vector<lite::Tensor *> input_tensors; in IncrementMatmulKernelCreate() local
/third_party/mindspore/mindspore/lite/src/delegate/nnrt/
Dnnrt_model_kernel.cc99 auto input_tensors = this->inputs(); in PrepareInputs() local
Dnnrt_delegate.cc154 auto input_tensors = model->inputs(); in PrepareInputs() local
/third_party/mindspore/mindspore/lite/tools/converter/legacy_optimizer/graph/
Dinfershape_pass.cc49 void FreeTensors(std::vector<Tensor *> *input_tensors, std::vector<Tensor *> *output_tensors) { in FreeTensors()
249 void PrintTensorShape(const std::vector<Tensor *> &input_tensors, const std::vector<Tensor *> &outp… in PrintTensorShape()
511 auto input_tensors = ConvertTensorToLiteTensor(graph, node->inputIndex); in InferSubgraph() local
/third_party/mindspore/mindspore/ccsrc/vm/
Dbackend.cc472 std::vector<tensor::TensorPtr> *input_tensors) { in CompileGraph()
799 std::vector<std::vector<tensor::TensorPtr>> input_tensors; in RunGraph() local
985 std::vector<std::vector<tensor::TensorPtr> *> input_tensors; in ConstructGraphCompilerInfo() local
993 const std::vector<tensor::TensorPtr> *input_tensors, bool need_erase) { in ConstructGraphCompilerInfo()
1038const std::vector<tensor::TensorPtr> *input_tensors, VectorRef *outputs) { in RunGraph()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/pyfunc/
Dpy_func_cpu_kernel.cc171 const std::vector<tensor::TensorPtr> &input_tensors) { in RawMemoryToPyObjects()
/third_party/mindspore/mindspore/lite/micro/coder/
Dgraph.cc191 std::vector<lite::Tensor *> CoderGraph::input_tensors() const { return input_tensors_; } in input_tensors() function in mindspore::lite::micro::CoderGraph
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dgraph_compiler.cc386std::vector<TensorPtr> *const input_tensors, bool *single_op_cache_hit, in CompileGraph()
478 …GetSingleOpRunInfoAndGraphInfo(const CNodePtr &kernel, const std::vector<TensorPtr> &input_tensors, in GetSingleOpRunInfoAndGraphInfo()
/third_party/mindspore/mindspore/lite/src/
Dlite_kernel_util.cc87 std::vector<lite::Tensor *> input_tensors; in SubgraphInputTensors() local
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dconstant_folding_fusion.cc350 auto input_tensors = GetCNodeInputTensors(input_cnode, fmk_type_); in Process() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_runtime.cc221 const std::vector<tensor::TensorPtr> &input_tensors) { in RunOpMallocPre()
315 void KernelRuntime::RunOpAssignMemory(const std::vector<tensor::TensorPtr> &input_tensors, in RunOpAssignMemory()
389 void KernelRuntime::RunOpAssignInputMemory(const std::vector<tensor::TensorPtr> &input_tensors, in RunOpAssignInputMemory()
1387 auto &input_tensors = graph.input_tensors(); in InitGraphInputTensors() local
/third_party/mindspore/mindspore/ccsrc/pipeline/pynative/
Dpynative_execute.cc308 …ngleOpGraphInfo(const OpExecInfoPtr &op_exec_info, const std::vector<tensor::TensorPtr> &input_ten… in GetSingleOpGraphInfo()
411 std::vector<tensor::TensorPtr> *input_tensors) { in PlantTensorTupleToVector()
425 …ertValueTupleToTensor(const py::object &input_object, std::vector<tensor::TensorPtr> *input_tensor… in ConvertValueTupleToTensor()
440std::vector<tensor::TensorPtr> *input_tensors, int64_t *const tensor_mask) { in ConvertMultiPyObjectToTensor()
461std::vector<tensor::TensorPtr> *input_tensors, int64_t *const tensor_mask) { in ConvertPyObjectToTensor()
498 std::vector<tensor::TensorPtr> *input_tensors) { in ConstructInputTensor()
1943 std::vector<tensor::TensorPtr> input_tensors; in RunOpInMs() local
/third_party/mindspore/mindspore/lite/src/cxx_api/model/
Dmodel_impl.cc209 auto input_tensors = session_->GetInputs(); in Predict() local

12