Home
last modified time | relevance | path

Searched defs:graph_inputs (Results 1 – 18 of 18) sorted by relevance

/third_party/mindspore/mindspore/lite/micro/coder/
Dcontext.h55 void set_graph_inputs(const std::vector<Tensor *> &graph_inputs) { graph_inputs_ = graph_inputs; } in set_graph_inputs()
58 std::vector<Tensor *> graph_inputs() const { return graph_inputs_; } in graph_inputs() function
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dspecify_graph_input_format.cc51 auto graph_inputs = graph->get_inputs(); in HandleGraphInput() local
Dtranspose_strategy.cc389 auto graph_inputs = func_graph->get_inputs(); in CanFusionIfInsert() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/
Dunused_node_remove_pass.cc56 auto graph_inputs = func_graph->get_inputs(); in ProcessGraph() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/
Dtrt_converter_context.cc99 const std::vector<AnfNodePtr> graph_inputs = func_graph_->parameters(); in InitInputTable() local
222 std::unordered_map<std::string, AnfNodePtr> graph_inputs; in GetGraphInputs() local
Dgraph_converter.cc49 …tNode(const FuncGraphPtr &graph, const std::string &model_data, const AnfNodePtrList &graph_inputs, in NewTrtNode()
166 std::vector<AnfNodePtr> graph_inputs; in RemoveParameterWithoutUser() local
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dgraph_compiler.cc60 std::vector<AnfNodePtr> graph_inputs = graph->inputs(); in CreateParameterDeviceAddress() local
462 const std::vector<TensorPtr> &graph_inputs, in GetSingleOpInputTensors()
471 const std::vector<TensorPtr> &graph_inputs, in GetSingleOpInputTensorByIndex()
Dcontrol_node_parser.cc1447 std::vector<AnfNodePtr> graph_inputs; in FetchBackendInputNode() local
/third_party/mindspore/mindspore/lite/micro/coder/generator/component/
Dcommon_component.cc190 std::vector<Tensor *> graph_inputs = ctx->graph_inputs(); in CodeGraphQuantArgsImplement() local
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dsession_basic.cc608 auto graph_inputs = graph->MutableInputs(); in CreateParameterFromTuple() local
636 auto graph_inputs = graph->MutableInputs(); in CreateNewParameterFromParameter() local
842 auto graph_inputs = kernel_graph->MutableInputs(); in ProcessNodeRetFunc() local
1406const std::vector<tensor::TensorPtr> &graph_inputs) { in GetParameterOutputTensor()
1435 const std::vector<tensor::TensorPtr> &graph_inputs, in GetOpInputTensors()
1471const std::vector<tensor::TensorPtr> &graph_inputs, in GetOpInputTensorByIndex()
1538 auto graph_inputs = graph->MutableInputs(); in ConstructKernelGraph() local
1588 auto graph_inputs = graph->MutableInputs(); in AddParameterToGraphInputs() local
Dcpu_session.cc68 auto graph_inputs = graph->MutableInputs(); in CreateNewParameterFromParameter() local
Dsession_basic.h243 const std::vector<tensor::TensorPtr> &graph_inputs, in BuildOpsInGraph()
Dascend_session.cc1081 const std::vector<tensor::TensorPtr> &graph_inputs, in GetOpInputStubTensors()
1118 const std::vector<tensor::TensorPtr> &graph_inputs, in BuildOpsInGraph()
1738 std::vector<AnfNodePtr> graph_inputs = to_graph->inputs(); in SyncInitialTenosrToDevice() local
/third_party/mindspore/mindspore/lite/tools/converter/
Danf_transform.cc225 auto graph_inputs = old_graph->get_inputs(); in RunParallelPass() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/
Donnx_model_parser.cc670std::vector<AnfNodePtr> *graph_inputs, const std::string &root_node_name) { in ConvertNodes()
846std::vector<AnfNodePtr> *graph_inputs, ops::PrimitiveC *primitive_c, in BuildCNode()
/third_party/mindspore/mindspore/ccsrc/vm/
Dbackend.cc513const std::vector<tensor::TensorPtr> &graph_inputs, InputTensorInfo *input_tensor_info, in GetControlOpInput()
603const std::vector<tensor::TensorPtr> &graph_inputs, InputTensorInfo *input_tensor_info, in RunControlOperator()
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_runtime.cc51 auto graph_inputs = graph.inputs(); in GetGraphInputs() local
552 auto graph_inputs = GetGraphInputs(graph); in AssignStaticMemoryInput() local
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/
Dconvert.cc428 std::vector<Operator> graph_inputs; in BuildSaveCheckpointGraph() local