Searched refs:CreateInt32Tensor (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | onehot_info.cc | 182 …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()
185 auto equal = gen_g.PushBack({gen_g.NewOpInst(EQUAL), floor_div, CreateInt32Tensor(mod_rank_)}); in ComputeReplaceGraph()
188 auto tensor_add = gen_g.PushBack({gen_g.NewOpInst(ADD), mul2, CreateInt32Tensor(1)}); in ComputeReplaceGraph()
190 auto sub2 = gen_g.PushBack({gen_g.NewOpInst(SUB), mul3, CreateInt32Tensor(1)}); 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()
134 …auto minimum = gen_g.PushBack({gen_g.NewOpInst(MINIMUM), relu, CreateInt32Tensor(slice_size - 1)}); 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()
774 …auto sub = gen_g.PushBack({gen_g.NewOpInst(SUB), gen_g.virtual_input_node(), CreateInt32Tensor(bia… in ComputeReplaceGraph()
776 …auto minimum = gen_g.PushBack({gen_g.NewOpInst(MINIMUM), relu, CreateInt32Tensor(slice_size_ - 1)}… in ComputeReplaceGraph()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | generate_graph.h | 42 AnfNodePtr CreateInt32Tensor(int64_t value);
|
| D | generate_graph.cc | 81 AnfNodePtr CreateInt32Tensor(int64_t value) { in CreateInt32Tensor() function
|