Searched defs:perm_node (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/format_type/ |
| D | replace_transdata_with_transpose.cc | 72 auto perm_node = NewValueNode(perm_value); in CreatePermValueNode() local
114 auto perm_node = CreatePermValueNode(func_graph, node); in CreateNewTranspose() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/ |
| D | unused_transpose_node_remove_pass.cc | 44 auto perm_node = node->input(2); in GetTransposePerm() local
|
| D | slice_prepose_pass.cc | 161 auto perm_node = node->input(2); in GetTransposePerm() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/add_attr_to_node/ |
| D | extract_volume_patches_fusion.cc | 33 auto perm_node = CreateTensorInput(kernel_graph, NewValueNode(perm_list)); in InsertTransposeNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/ |
| D | transpose_fusion.cc | 231 auto perm_node = BuildIntVecParameterNode(func_graph, perm, name + "_perm"); in TransTransFusion() local
326 auto perm_node = transpose_cnode->input(kInputIndexTwo); in Process() local
|
| D | flash_attention_fusion.cc | 2252 …auto perm_node = BuildIntVecParameterNode(func_graph, new_perm, k_trans_BNDS->fullname_with_scope(… in CreateFlashAttentionNodeForSDPreMul() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/ |
| D | transpose_matmul_fusion.cc | 34 auto perm_node = transpose->input(kIndex2); in IsTargetTranspose() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/ |
| D | insert_format_transform_op.cc | 122 auto perm_node = CreateTupleValue(kernel_graph, transpose_perm); in InsertTransposeOp() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/common/ |
| D | anf_util.cc | 615 auto perm_node = BuildIntVecParameterNode(func_graph, perm, cnode_name + "_perm"); in GenTransposeNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/core/ |
| D | transform_op_optimizer.cc | 330 auto perm_node = gb.Value(perm_tensor); in SetInput() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/model/ |
| D | op_node.cc | 493 auto perm_node = inputs[1]; in InferFormat() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/common/ |
| D | gllo_utils.cc | 1143 auto perm_node = BuildIntVecParameterNode(func_graph, perm, cnode_name + "_perm"); in GenTransposeNode() local
|