Searched refs:attr_shape (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | broadcast_to_info.cc | 163 Attr attr_shape = std::make_pair(SHAPE, MakeValue(to_shape)); in ComputeReplaceGraph() local
164 OperatorAttrs attrs = {attr_shape}; in ComputeReplaceGraph()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | dynamic_gru_v2_grad_fission.cc | 404 …std::vector<int64_t> attr_shape = {SizeToLong(t_size), SizeToLong(input_size), SizeToLong(kGateNum… in CreateWBroadcastToDNode() local
405 AnfAlgo::SetNodeAttr(kAttrShape, MakeValue(attr_shape), broadcast_to_d); in CreateWBroadcastToDNode()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/ |
| D | pipeline_transformer.cc | 650 Attr attr_shape = std::make_pair(SHAPE, shape_type_pair.first); in InsertReceive() local
654 OperatorAttrs attrs = {attr_tag, attr_rank, attr_shape, attr_dtype, attr_group, attr_group_back}; in InsertReceive()
|