Searched refs:kAttrShape (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | reshape_transpose_fusion.cc | 85 AnfAlgo::SetNodeAttr(kAttrShape, MakeValue(Convert2Long(reshape_output_shape)), new_node); in Process()
|
| D | transpose_reshape_fusion.cc | 84 AnfAlgo::SetNodeAttr(kAttrShape, MakeValue(Convert2Long(reshape_output_shape)), new_node); in Process()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/ |
| D | communication_op_fusion.cc | 379 if (AnfAlgo::HasNodeAttr(kAttrShape, final_node)) { in CreateFusedCommunicationOp()
381 AnfAlgo::SetNodeAttr(kAttrShape, MakeValue(fusion_total_shape), fused_node); in CreateFusedCommunicationOp()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | cdist_fission.cc | 78 AnfAlgo::SetNodeAttr(kAttrShape, MakeValue(shape), broadcast_to); in AddBroadCastToNode()
|
| D | dynamic_gru_v2_grad_fission.cc | 405 AnfAlgo::SetNodeAttr(kAttrShape, MakeValue(attr_shape), broadcast_to_d); in CreateWBroadcastToDNode()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hccl_adapter/ |
| D | converter.cc | 207 (void)ConvertAttr<std::vector<int64_t>>(cnode, op_desc, kAttrShape, ge::HCOM_ATTR_SHAPE); in GenerateStubGeNode()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | utils.h | 343 constexpr auto kAttrShape = "shape"; variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ |
| D | ascend_helper.cc | 51 AnfAlgo::SetNodeAttr(kAttrShape, MakeValue(dst_shape), reshape); in CreateReshapeNode()
|