Home
last modified time | relevance | path

Searched defs:SetOutputs (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/model/
Dlite_graph.h38 void SetOutputs(const NodePtrList &nodes) { output_->SetInputs(nodes); } in SetOutputs() function
69 void SetOutputs(const NodePtrList &nodes) const { graph_->output_->SetInputs(nodes); } in SetOutputs() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/
Dexecution_flow.h43 void SetOutputs(const std::vector<InferTensor *> &outputs) override { outputs_ = outputs; } in SetOutputs() function
Dexecution_plan.h53 void SetOutputs(const std::vector<abstract::Tensor *> &outputs) override { outputs_ = outputs; } in SetOutputs() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/nnrt/
Dnnrt_model_kernel.cc143 int NNRTModelKernel::SetOutputs() { in SetOutputs() function in mindspore::NNRTModelKernel
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/device/
Dhccl_adapter_test.cc47 …void SetOutputs(const CNodePtr &cnode, const std::vector<ShapeVector> &shape, const std::vector<Ty… in SetOutputs() function in mindspore::hccl::TestHcclAdapter
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/hardware/
Dge_graph_executor.cc1125 void SetOutputs(const std::vector<KernelWithIndex> &graph_outputs, in SetOutputs() function