Home
last modified time | relevance | path

Searched defs:output_tensors (Results 1 – 25 of 37) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/
Dtest_square_sum_all.py36 def run_net(datatype, input_tensors, output_tensors): argument
82 def test_cpu(dtype, input_tensors, output_tensors): argument
/third_party/mindspore/mindspore-src/source/mindspore/lite/providers/dpico/manager/
Dacl_model_manager.cc240 int AclModelManager::CopyAclOutputsToTensorData(const std::vector<mindspore::MSTensor> &output_tens… in CopyAclOutputsToTensorData()
361 const std::vector<mindspore::MSTensor> &output_tensors) { in Init()
463 int AclModelManager::PrepareAclOutputs(std::vector<mindspore::MSTensor> *output_tensors) { in PrepareAclOutputs()
563 int AclModelManager::UpdateAclOutputs(std::vector<mindspore::MSTensor> *output_tensors) { in UpdateAclOutputs()
596 const std::vector<mindspore::MSTensor> &output_tensors, in Execute()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/
Dop_coder.cc32 const std::vector<Tensor *> OperatorCoder::output_tensors() const { return output_tensors_; } in output_tensors() function in mindspore::lite::micro::OperatorCoder
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/
Dutils_test.cc63 auto output_tensors = kernel::KernelExecUtil::SubgraphOutputTensors(kernels); in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/legacy_optimizer/graph/
Dinfershape_pass.cc50 void FreeTensors(std::vector<Tensor *> *input_tensors, std::vector<Tensor *> *output_tensors) { in FreeTensors()
294 …ensorShape(const std::vector<Tensor *> &input_tensors, const std::vector<Tensor *> &output_tensors in PrintTensorShape()
314 int SetDataType(MetaGraphT *graph, const std::vector<Tensor *> &output_tensors, in SetDataType()
364 int CopyOutputInfoToTensorT(MetaGraphT *graph, const std::vector<Tensor *> &output_tensors, in CopyOutputInfoToTensorT()
609 auto output_tensors = ConvertTensorToLiteTensor(graph, node->outputIndex); in InferSubgraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/
Dhook.cc36 ValuePtrList output_tensors; in GetCValue() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/
Dtensorrt_graph_executor.cc124 std::vector<TensorInfo> *output_tensors) { in GetAbstractArgsFromCNode()
204 std::vector<TensorInfo> *output_tensors) { in GetModelOutputsInfo()
420 std::vector<TensorInfo> output_tensors; in BuildSubGraph() local
459 const std::vector<TensorInfo> &output_tensors) { in FindTensorRTOp()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/control_flow/actor/
Dswitch_actor.cc86 auto output_tensors = kernel_->out_tensors(); in AppendOutputTensors() local
124 auto output_tensors = kernel_->out_tensors(); in UpdateActorOutput() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/const_fold/
Dfold_utils.cc156 …Node(const FuncGraphPtr &func_graph, const CNodePtr &cnode, std::vector<Tensor *> output_tensors) { in ReplaceCNode()
289 std::vector<lite::Tensor *> output_tensors; in DoConstantFold() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/utils/
Druntime_utils.cc100 auto output_tensors = AnfAlgo::GetOrCreateAllOutputKernelTensors(kernel_node); in UpdateKernelNodeOutputInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/device/
Dlaunch_transdata.cc142 std::vector<kernel::KernelTensorPtr> output_tensors; in LaunchOpKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/kernel/ascend_native/
Dascend_native_composite_kernel.cc78 std::vector<kernel::InferTensor *> output_tensors; in CreateKernel() local
225std::vector<kernel::InferTensor *> *output_tensors) { in CreateOutputKernelTensors()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/ascend_native/
Ddelegate.cc216std::vector<kernel::InferTensor *> *output_tensors, in CreateOutputKernelTensors()
250 std::vector<kernel::InferTensor *> output_tensors; in CreateKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/hardware/
Dgpu_somas.cc116 auto output_tensors = node->output_tensors_; in InplaceNodeProcess() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/hardware/
Dacl_somas.cc115 auto output_tensors = node->output_tensors_; in RuntimeNodeProcess() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/
Doutput_actor.cc29 void UpdateOutputTensorShape(const std::vector<TensorPtr> &output_tensors, in UpdateOutputTensorShape()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/session/
Dlite_infer_session.cc209 std::vector<MutableTensorImplPtr> output_tensors; in GetOutputs() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/graph_compiler/
Dbackend_base.cc1045 auto &output_tensors = actor_set->output_actor_->outputs(); in ConstructOutputs() local
1205const std::vector<tensor::TensorPtr> &output_tensors, in ConstructOutputByAbstract()
1361 …ol IsEmptySequence(const AnfNodePtr &output_node, const std::vector<tensor::TensorPtr> &output_ten… in IsEmptySequence()
1390const std::vector<tensor::TensorPtr> &output_tensors, size_t *output_position, in ConstructOutputs()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/pyboost/
Dpyboost_utils.cc601 std::vector<tensor::BaseTensorPtr> output_tensors; in CastTensor() local
612 std::vector<tensor::BaseTensorPtr> output_tensors; in CastTensor() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/
Dkernel_exec_util.cc175 std::vector<lite::Tensor *> output_tensors; in SubgraphOutputTensors() local
372 std::vector<lite::Tensor *> output_tensors; in CreateSubGraphKernel() local
Dlite_mindrt.cc313 auto output_tensors = this->kernel_->out_tensors(); in CompileArrowThroughOutputTensors() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/
Dgraph.cc203 std::vector<lite::Tensor *> CoderGraph::output_tensors() const { return output_tensors_; } in output_tensors() function in mindspore::lite::micro::CoderGraph
/third_party/mesa3d/src/gallium/drivers/etnaviv/
Detnaviv_ml.h92 unsigned output_tensors[MAX_TENSORS]; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/
Dbase.h60 std::vector<tensor::BaseTensorPtr> output_tensors; member
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/graph_executor/litert/
Dgraph_executor.cc303 std::vector<tensor::Tensor> output_tensors; in GetOutputInfos() local

12