Searched refs:CreateValueNode (Results 1 – 9 of 9) sorted by relevance
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
D | lin_space_fission.cc | 74 ValueNodePtr CreateValueNode(const AnfNodePtr &node) { in CreateValueNode() function 109 auto assist_const = CreateValueNode(cnode); in Process()
|
D | max_pool3d_grad_grad_fission.cc | 76 ValueNodePtr CreateValueNode(const AnfNodePtr &node) { in CreateValueNode() function 112 auto assist_const = CreateValueNode(cnode); in Process()
|
D | space_to_depth_split.cc | 73 ValueNodePtr CreateValueNode(const AnfNodePtr &node) { in CreateValueNode() function 116 auto last_input_value = CreateValueNode(cnode); in Process()
|
D | topk_split.cc | 65 ValueNodePtr CreateValueNode() { in CreateValueNode() function 168 auto indices_const = CreateValueNode(); in Process()
|
D | dynamic_rnn_grad_fission_v2.cc | 482 AnfNodePtr CreateValueNode(const FuncGraphPtr &func_graph, const CNodePtr &dynamic_rnn_grad_cnode) { in CreateValueNode() function 567 auto value_node = CreateValueNode(func_graph, dynamic_rnn_grad_cnode); in Process()
|
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
D | sparse_softmax_cross_entropy_with_logits_unify_mindir.cc | 37 ValueNodePtr CreateValueNode(const ValuePtr &value_ptr, TypeId output_type) { in CreateValueNode() function 69 auto value_on_node = CreateValueNode(value_on, kNumberTypeFloat32); in CreateOneHot() 72 auto value_off_node = CreateValueNode(value_off, kNumberTypeFloat32); in CreateOneHot() 154 auto axis_node = CreateValueNode(MakeValue(range), kNumberTypeInt64); in GetAxisNode() 267 auto multiples_node = CreateValueNode(multiples, kNumberTypeInt64); in CreateTile() 311 auto y_node = CreateValueNode(y, kNumberTypeFloat32); in CreateRealDiv() 366 auto y_node = CreateValueNode(tensor_y, kNumberTypeFloat32); in CreateMul()
|
/third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/ |
D | onnx_model_parser.cc | 97 ValueNodePtr CreateValueNode(const schema::PrimitiveType &op_type) { in CreateValueNode() function 117 auto sub_value_node = CreateValueNode(schema::PrimitiveType_SubFusion); in AddIterNumsUpdateEdge() 391 auto less_value_node = CreateValueNode(schema::PrimitiveType_Less); in BuildCondGraph() 408 auto and_value_node = CreateValueNode(schema::PrimitiveType_LogicalAnd); in BuildCondGraph() 1163 auto add_value_node = CreateValueNode(schema::PrimitiveType_AddFusion); in AddTensorArrayEdge() 1224 auto set_item_value_node = CreateValueNode(schema::PrimitiveType_TensorListSetItem); in AddTensorArrayEdge()
|
/third_party/mindspore/mindspore/ccsrc/backend/session/ |
D | session_basic.h | 174 std::vector<AnfNodePtr> CreateValueNode(const CNodePtr &cnode, KernelGraph *graph);
|
D | session_basic.cc | 974 std::vector<AnfNodePtr> SessionBasic::CreateValueNode(const CNodePtr &cnode, KernelGraph *graph) { in CreateValueNode() function in mindspore::session::SessionBasic 1035 cnode_inputs = CreateValueNode(cnode, graph); in CreateNewCNode()
|