Searched defs:node_input (Results 1 – 7 of 7) sorted by relevance
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
D | reorder_ops.cc | 104 auto node_input = AnfAlgo::GetInputNode(new_node, 0); in SetNodeInfo() local 217 auto node_input = AnfAlgo::GetInputNode(node, 0); in ReorderTypeInsensitiveCastDown() local 274 auto node_input = AnfAlgo::GetInputNode(node, index); in ReorderCastUpTypeInsensitive() local
|
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
D | special_op_eliminate.h | 294 std::vector<AnfNodePtr> node_input = in operator() local 333 std::vector<AnfNodePtr> node_input = in operator() local
|
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
D | dropout_unify_mindir.cc | 125 CNodePtr CreateDynamicShapeCNode(const FuncGraphPtr &func_graph, const AnfNodePtr &node_input, in CreateDynamicShapeCNode()
|
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/ |
D | program_specialize.cc | 334 auto node_input = old_inputs[i]; in ProcessNode() local
|
/third_party/mindspore/mindspore/ccsrc/backend/session/ |
D | ascend_auto_monad.cc | 894 auto node_input = node->input(i); in FindInputNode() local 919 CNodePtr InsertStackPop(const AnfNodePtr &node_input, std::vector<CNodePtr> *stack_pushs) { in InsertStackPop()
|
D | session_basic.cc | 749 CNodePtr SessionBasic::CreateSwitchInput(const CNodePtr &cnode, const AnfNodePtr &node_input, Kerne… in CreateSwitchInput() 1007 auto node_input = cnode->input(index); in CreateCNodeInputs() local
|
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
D | step_parallel.cc | 205 std::vector<AnfNodePtr> node_input; in InsertNode() local 239 std::vector<AnfNodePtr> node_input; in ReplaceNode() local 672 std::vector<AnfNodePtr> node_input = CreateInput(op, value_node, SPLIT_TENSOR); in SplitTensorList() local
|