Searched refs:kAttrIsDynamicShape (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | layer_norm_grad_split.cc | 53 AnfAlgo::SetNodeAttr(kAttrIsDynamicShape, MakeValue(true), layer_norm_x_backprop); in CreateOutputsOfLayerNormXBackpropV2()
77 AnfAlgo::SetNodeAttr(kAttrIsDynamicShape, MakeValue(true), layer_norm_beta_gamma_backprop); in CreateOutputsOfLayerNormBetaGammaBackpropV2()
|
| D | gather_v2_ds_fission.cc | 80 AnfAlgo::SetNodeAttr(kAttrIsDynamicShape, MakeValue(true), pad); in CreatePad()
104 AnfAlgo::SetNodeAttr(kAttrIsDynamicShape, MakeValue(true), gather_v2); in CreateGatherV2Ds()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/ |
| D | insert_tensor_move_for_getnext.cc | 47 AnfAlgo::SetNodeAttr(kAttrIsDynamicShape, MakeValue(true), new_node); in InsertTensorMoveForGetNextOutputs()
|
| D | insert_tensor_move_for_cascade.cc | 88 AnfAlgo::SetNodeAttr(kAttrIsDynamicShape, MakeValue(true), tensor_move); in InsertTensorMove()
|
| D | insert_tensor_move_for_hccl_op.cc | 125 AnfAlgo::SetNodeAttr(kAttrIsDynamicShape, MakeValue(true), tensor_move); in InsertTensorMove()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_dynaminc_shape_util.cc | 335 AnfAlgo::SetNodeAttr(kAttrIsDynamicShape, MakeValue(is_dyanmic_shape), cnode); in SetDynamicShapeAttr()
350 auto is_dynamic_shape = AnfAlgo::HasNodeAttr(kAttrIsDynamicShape, cnode); in GetDynamicShapeAttr()
354 is_dynamic_shape = AnfAlgo::GetNodeAttr<bool>(cnode, kAttrIsDynamicShape); in GetDynamicShapeAttr()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ |
| D | ascend_helper.cc | 314 AnfAlgo::SetNodeAttr(kAttrIsDynamicShape, MakeValue(true), trans_node); in NewTransOpNode()
356 AnfAlgo::SetNodeAttr(kAttrIsDynamicShape, MakeValue(true), cast); in AddCastOpNodeToGraph()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hardware/cpu/ |
| D | cpu_device_context.cc | 137 AnfAlgo::SetNodeAttr(kAttrIsDynamicShape, MakeValue(true), cnode); in UpdateGraphDynamicShapeAttr()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hardware/gpu/ |
| D | gpu_device_context.cc | 298 AnfAlgo::SetNodeAttr(kAttrIsDynamicShape, MakeValue(true), cnode); in UpdateGraphDynamicShapeAttr()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | utils.h | 445 constexpr auto kAttrIsDynamicShape = "is_dynamic_shape"; variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | anf_runtime_algorithm.cc | 1983 get_bool_attr(prim, kAttrIsDynamicShape); in HasDynamicShapeFlag()
1988 GetBooleanAttr(node, kAttrIsDynamicShape); in IsDynamicShape()
|
| D | session_basic.cc | 2348 AnfAlgo::SetNodeAttr(kAttrIsDynamicShape, MakeValue(true), cnode); in UpdateGraphDynamicShapeAttr()
|