Home
last modified time | relevance | path

Searched refs:GetGraphInputs (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/
Dtrt_converter_context.h62 std::vector<AnfNodePtr> GetGraphInputs() const;
Dgraph_converter.cc153 const AnfNodePtrList &useful_parameters = converter->GetGraphInputs(); in BuildTrtNode()
Dtrt_converter_context.cc220 std::vector<AnfNodePtr> TrtConverterContext::GetGraphInputs() const { in GetGraphInputs() function in mindspore::opt::TrtConverterContext
/third_party/mindspore/mindspore/lite/tools/common/
Dtensor_util.h128 const std::vector<int> &GetGraphInputs() const { return graphInputs; } in GetGraphInputs() function
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_runtime.cc50 std::vector<AnfNodePtr> GetGraphInputs(const session::KernelGraph &graph) { in GetGraphInputs() function
552 auto graph_inputs = GetGraphInputs(graph); in AssignStaticMemoryInput()