Home
last modified time | relevance | path

Searched refs:NewOpInst (Results 1 – 11 of 11) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dunique_info.cc132 …auto sub = gen_g.PushBack({gen_g.NewOpInst(SUB), gen_g.virtual_input_node(), CreateInt32Tensor(bia… in ComputeReplaceGraph()
133 auto relu = gen_g.PushBack({gen_g.NewOpInst(RELU), sub}); in ComputeReplaceGraph()
134 …auto minimum = gen_g.PushBack({gen_g.NewOpInst(MINIMUM), relu, CreateInt32Tensor(slice_size - 1)}); in ComputeReplaceGraph()
135 auto equal = gen_g.PushBack({gen_g.NewOpInst(EQUAL), sub, minimum}); in ComputeReplaceGraph()
136 auto unique = gen_g.PushBack({gen_g.NewOpInst(replace_op_name_), gen_g.virtual_input_node()}); in ComputeReplaceGraph()
139 …auto tuple_getitem_0 = gen_g.PushBack({gen_g.NewOpInst(tuple_getitem_op), unique, CreatInt64Imm(0)… in ComputeReplaceGraph()
140 …auto tuple_getitem_1 = gen_g.PushBack({gen_g.NewOpInst(tuple_getitem_op), unique, CreatInt64Imm(1)… in ComputeReplaceGraph()
141 auto dtype = gen_g.PushBack({gen_g.NewOpInst(DTYPE), tuple_getitem_1}); in ComputeReplaceGraph()
142 auto cast = gen_g.PushBack({gen_g.NewOpInst(CAST), equal, dtype}); in ComputeReplaceGraph()
143 auto mul = gen_g.PushBack({gen_g.NewOpInst(MUL), tuple_getitem_1, cast}); in ComputeReplaceGraph()
[all …]
Donehot_info.cc182 …gen_g.PushBack({gen_g.NewOpInst(FLOORDIV), gen_g.virtual_input_node(), CreateInt32Tensor(classes_e… in ComputeReplaceGraph()
183 …auto mul1 = gen_g.PushBack({gen_g.NewOpInst(MUL), floor_div, CreateInt32Tensor(classes_each_device… in ComputeReplaceGraph()
184 auto sub1 = gen_g.PushBack({gen_g.NewOpInst(SUB), gen_g.virtual_input_node(), mul1}); in ComputeReplaceGraph()
185 auto equal = gen_g.PushBack({gen_g.NewOpInst(EQUAL), floor_div, CreateInt32Tensor(mod_rank_)}); in ComputeReplaceGraph()
186 auto cast = gen_g.PushBack({gen_g.NewOpInst(CAST), equal, CreatTypeInt(32)}); in ComputeReplaceGraph()
187 auto mul2 = gen_g.PushBack({gen_g.NewOpInst(MUL), sub1, cast}); in ComputeReplaceGraph()
188 auto tensor_add = gen_g.PushBack({gen_g.NewOpInst(ADD), mul2, CreateInt32Tensor(1)}); in ComputeReplaceGraph()
189 auto mul3 = gen_g.PushBack({gen_g.NewOpInst(MUL), cast, tensor_add}); in ComputeReplaceGraph()
190 auto sub2 = gen_g.PushBack({gen_g.NewOpInst(SUB), mul3, CreateInt32Tensor(1)}); in ComputeReplaceGraph()
193 …auto onehot = gen_g.PushBack({gen_g.NewOpInst(ONEHOT, attrs_onehot), sub2, CreatInt64Imm(classes_e… in ComputeReplaceGraph()
Dunsorted_segment_op_info.cc282 …auto segment_min = gen_g.PushBack({gen_g.NewOpInst(UNSORTED_SEGMENT_MIN), gen_g.virtual_input_node… in ComputeReplaceGraph()
284 …auto expandim_output = gen_g.PushBack({gen_g.NewOpInst(EXPAND_DIMS), segment_min, CreatInt64Imm(0)… in ComputeReplaceGraph()
285 auto all_gather_output = gen_g.PushBack({gen_g.NewOpInst(ALL_GATHER), expandim_output}); in ComputeReplaceGraph()
286 …auto final_output = gen_g.PushBack({gen_g.NewOpInst(REDUCE_MIN), all_gather_output, CreatInt64Imm(… in ComputeReplaceGraph()
316 …auto segment_max = gen_g.PushBack({gen_g.NewOpInst(UNSORTED_SEGMENT_MAX), gen_g.virtual_input_node… in ComputeReplaceGraph()
318 …auto expandim_output = gen_g.PushBack({gen_g.NewOpInst(EXPAND_DIMS), segment_max, CreatInt64Imm(0)… in ComputeReplaceGraph()
319 auto all_gather_output = gen_g.PushBack({gen_g.NewOpInst(ALL_GATHER), expandim_output}); in ComputeReplaceGraph()
320 …auto final_output = gen_g.PushBack({gen_g.NewOpInst(REDUCE_MAX), all_gather_output, CreatInt64Imm(… in ComputeReplaceGraph()
Dgather_v2_p_info.cc760 …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()
775 auto relu = gen_g.PushBack({gen_g.NewOpInst(RELU), sub}); in ComputeReplaceGraph()
776 …auto minimum = gen_g.PushBack({gen_g.NewOpInst(MINIMUM), relu, CreateInt32Tensor(slice_size_ - 1)}… in ComputeReplaceGraph()
777 auto equal = gen_g.PushBack({gen_g.NewOpInst(EQUAL), sub, minimum}); in ComputeReplaceGraph()
779 …gen_g.PushBack({gen_g.NewOpInst(replace_op_name_), gen_g.virtual_input_node(), minimum, CreatInt64… in ComputeReplaceGraph()
780 auto dtype = gen_g.PushBack({gen_g.NewOpInst(DTYPE), gather_v2}); in ComputeReplaceGraph()
781 auto cast = gen_g.PushBack({gen_g.NewOpInst(CAST), equal, dtype}); in ComputeReplaceGraph()
782 auto expand_dims = gen_g.PushBack({gen_g.NewOpInst(EXPAND_DIMS), cast, CreatInt64Imm(axis_ - 1)}); in ComputeReplaceGraph()
[all …]
Dconv2d_info.cc711 return gen_g_.PushBack({gen_g_.NewOpInst(node_name, conv2d_attrs), new_input, cnode->input(2)}); in GenerateConv2DNode()
718 …return gen_g_.PushBack({gen_g_.NewOpInst(node_name, conv2d_attrs), new_input, cnode->input(2), cno… in GenerateConv2DNode()
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()
760 …auto alltoall_v = gen_g_.PushBack({gen_g_.NewOpInst(NEIGHBOREXCHANGE, alltoall_attrs), make_tuple_… in ComputeReplaceGraph()
773 auto concat_l = gen_g_.PushBack({gen_g_.NewOpInst(CONCAT, concat_attrs), make_tuple_l}); in ComputeReplaceGraph()
781 auto concat_r = gen_g_.PushBack({gen_g_.NewOpInst(CONCAT, concat_attrs), make_tuple_r}); in ComputeReplaceGraph()
795 auto concat_r_1 = gen_g_.PushBack({gen_g_.NewOpInst(CONCAT, concat_attrs), make_tuple_r_1}); in ComputeReplaceGraph()
Dbroadcast_to_info.cc165 …auto new_broadcast_to = gen_g.PushBack({gen_g.NewOpInst(BROADCAST_TO, attrs), gen_g.virtual_input_… in ComputeReplaceGraph()
Duniform_candidate_sampler_info.cc232 …auto new_sampler_op = gen_g.PushBack({gen_g.NewOpInst(UNIFORM_CANDIDATE_SAMPLER, attrs), gen_g.vir… in ComputeReplaceGraph()
Dslice_info.cc232 …auto slice = gen_g.PushBack({gen_g.NewOpInst(SLICE), gen_g.virtual_input_node(), new_begin, new_si… in ComputeReplaceGraph()
Dmatmul_dds_info.cc248 …gen_g.PushBack({gen_g.NewOpInst(MATMUL_DDS, attrs), gen_g.virtual_input_node(), gen_g.virtual_inpu… in ComputeReplaceGraph()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/
Dgenerate_graph.h54 AnfNodePtr NewOpInst(const OperatorName &op_name, const OperatorAttrs &attrs);
55 AnfNodePtr NewOpInst(const OperatorName &op_name);
Dgenerate_graph.cc174 AnfNodePtr GenerateGraph::NewOpInst(const OperatorName &op_name, const OperatorAttrs &attrs) { in NewOpInst() function in mindspore::parallel::GenerateGraph
184 AnfNodePtr GenerateGraph::NewOpInst(const OperatorName &op_name) { in NewOpInst() function in mindspore::parallel::GenerateGraph