Searched refs:virtual_input_node (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | matmul_dds_info.cc | 248 ….PushBack({gen_g.NewOpInst(MATMUL_DDS, attrs), gen_g.virtual_input_node(), gen_g.virtual_input_nod… in ComputeReplaceGraph()
249 gen_g.virtual_input_node(), gen_g.virtual_input_node()}); in ComputeReplaceGraph()
|
| D | unsorted_segment_op_info.cc | 282 …to segment_min = gen_g.PushBack({gen_g.NewOpInst(UNSORTED_SEGMENT_MIN), gen_g.virtual_input_node(), in ComputeReplaceGraph()
283 gen_g.virtual_input_node(), CreatInt64Imm(num_segments)}); in ComputeReplaceGraph()
316 …to segment_max = gen_g.PushBack({gen_g.NewOpInst(UNSORTED_SEGMENT_MAX), gen_g.virtual_input_node(), in ComputeReplaceGraph()
317 gen_g.virtual_input_node(), CreatInt64Imm(num_segments)}); in ComputeReplaceGraph()
|
| D | unique_info.cc | 132 …auto sub = gen_g.PushBack({gen_g.NewOpInst(SUB), gen_g.virtual_input_node(), CreateInt32Tensor(bia… in ComputeReplaceGraph()
136 auto unique = gen_g.PushBack({gen_g.NewOpInst(replace_op_name_), gen_g.virtual_input_node()}); in ComputeReplaceGraph()
|
| D | onehot_info.cc | 182 …gen_g.PushBack({gen_g.NewOpInst(FLOORDIV), gen_g.virtual_input_node(), CreateInt32Tensor(classes_e… in ComputeReplaceGraph()
184 auto sub1 = gen_g.PushBack({gen_g.NewOpInst(SUB), gen_g.virtual_input_node(), mul1}); in ComputeReplaceGraph()
|
| D | gather_v2_p_info.cc | 760 …gen_g.PushBack({gen_g.NewOpInst(SUB), gen_g.virtual_input_node(), CreateInt32Tensor(index_offset_)… in ComputeReplaceGraph()
762 …gen_g.PushBack({gen_g.NewOpInst(replace_op_name_), gen_g.virtual_input_node(), sub_node, CreatInt6… in ComputeReplaceGraph()
774 …auto sub = gen_g.PushBack({gen_g.NewOpInst(SUB), gen_g.virtual_input_node(), CreateInt32Tensor(bia… in ComputeReplaceGraph()
779 …gen_g.PushBack({gen_g.NewOpInst(replace_op_name_), gen_g.virtual_input_node(), minimum, CreatInt64… in ComputeReplaceGraph()
|
| D | broadcast_to_info.cc | 165 …_broadcast_to = gen_g.PushBack({gen_g.NewOpInst(BROADCAST_TO, attrs), gen_g.virtual_input_node()}); in ComputeReplaceGraph()
|
| D | uniform_candidate_sampler_info.cc | 232 … = gen_g.PushBack({gen_g.NewOpInst(UNIFORM_CANDIDATE_SAMPLER, attrs), gen_g.virtual_input_node()}); in ComputeReplaceGraph()
|
| D | slice_info.cc | 232 …auto slice = gen_g.PushBack({gen_g.NewOpInst(SLICE), gen_g.virtual_input_node(), new_begin, new_si… in ComputeReplaceGraph()
|
| D | conv2d_info.cc | 743 …auto slice_left = gen_g_.PushBack({gen_g_.NewOpInst(STRIDED_SLICE), gen_g_.virtual_input_node(), s… in ComputeReplaceGraph()
752 …auto slice_right = gen_g_.PushBack({gen_g_.NewOpInst(STRIDED_SLICE), gen_g_.virtual_input_node(), … in ComputeReplaceGraph()
790 …fNodePtr> make_tuple_r_inputs_1 = {NewValueNode(prim::kPrimMakeTuple), gen_g_.virtual_input_node(), in ComputeReplaceGraph()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | generate_graph.h | 53 AnfNodePtr virtual_input_node() { return virtual_input_node_; } in virtual_input_node() function
|