| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | remove_format_transform_pair.cc | 31 VectorRef transpose1 = VectorRef({prim::kPrimTranspose, X}); in DefinePattern()
32 VectorRef transpose2 = VectorRef({prim::kPrimTranspose, transpose1}); in DefinePattern()
43 if (AnfAlgo::GetCNodeName(node) != prim::kPrimTranspose->name() || in Process()
44 AnfAlgo::GetCNodeName(input_node) != prim::kPrimTranspose->name()) { in Process()
|
| D | remove_redundant_format_transform.cc | 30 VectorRef transpose = VectorRef({prim::kPrimTranspose, X}); in DefinePattern()
46 if (AnfAlgo::GetCNodeName(used_node) == prim::kPrimTranspose->name()) { in Process()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | transpose_fusion.cc | 37 auto is_transpose = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTranspose>); in DefineBNPattern()
57 auto is_transpose = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTranspose>); in DefineActivationscalePattern()
75 auto is_transpose = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTranspose>); in DefineActivationPattern()
89 auto is_transpose = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTranspose>); in DefineBiasAddPattern()
105 auto is_transpose1 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTranspose>); in DefineTransTransPattern()
107 auto is_transpose2 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTranspose>); in DefineTransTransPattern()
148 if (!CheckPrimitiveType(trans_cnode_2, prim::kPrimTranspose) || in TransTransFusion()
149 !CheckPrimitiveType(trans_cnode_2->input(1), prim::kPrimTranspose)) { in TransTransFusion()
|
| D | multi_head_attention_fusion.cc | 38 auto is_transpose = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTranspose)); in DefineEmbedding()
90 auto is_transpose = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTranspose)); in DefineMPWithMaskPattern()
107 auto is_tranpose = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTranspose)); in DefineDensePattern()
142 auto is_transpose = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTranspose)); in DefineProcessInputPattern()
151 auto is_transpose = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTranspose)); in DefineProcessOutputPattern()
|
| D | tflite_rel_pos_multi_head_attention_fusion.cc | 86 auto is_transpose1 = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTranspose)); in DefinePatterns()
101 auto is_transpose2 = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTranspose)); in DefinePatterns()
109 auto is_transpose3 = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTranspose)); in DefinePatterns()
411 auto is_transpose = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTranspose)); in DefineProcessInputPattern()
423 auto is_transpose = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTranspose)); in DefineProcessOutputPattern()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/graph/ |
| D | unused_transpose_node_remove_pass.cc | 34 if (!CheckPrimitiveType(node, prim::kPrimTranspose)) { in GetTransposePerm()
74 if (CheckPrimitiveType(node, prim::kPrimTranspose)) { in Run()
91 if (!CheckPrimitiveType(conv_node->input(kTransposeInput), prim::kPrimTranspose)) { in Run()
|
| D | redundant_op_remove_pass.cc | 35 if (CheckPrimitiveType(first_input, prim::kPrimTranspose)) { in ProcessInputIsMonad()
41 if (CheckPrimitiveType(second_input, prim::kPrimTranspose)) { in ProcessInputIsMonad()
86 if (CheckPrimitiveType(pre_node, prim::kPrimTranspose)) { in ProcessDependencyWithTwoNodes()
90 if (CheckPrimitiveType(post_node, prim::kPrimTranspose)) { in ProcessDependencyWithTwoNodes()
127 if (CheckPrimitiveType(cnode->input(0), prim::kPrimTranspose)) { in ProcessInputHaveDependency()
159 if (CheckPrimitiveType(anf_node, prim::kPrimTranspose)) { in ReplaceOp()
389 if (CheckPrimitiveType(node, prim::kPrimTranspose)) { in Run()
|
| D | decrease_transpose_algo.cc | 45 if (CheckPrimitiveType(cur_node, prim::kPrimTranspose)) { in FindAreaSurroundedByTranspose()
107 …if (!CheckPrimitiveType(cnode, prim::kPrimTranspose) || GetTransposePerm(cnode, &perm) != lite::RE… in SetTransType()
207 if (!CheckPrimitiveType(cnode, prim::kPrimTranspose)) { in PostTransposeFusion()
218 if (CheckPrimitiveType(post_node, prim::kPrimTranspose)) { in PostTransposeFusion()
249 if (CheckPrimitiveType(new_cnode_input, prim::kPrimTranspose)) { in GenNewInput()
421 if (CheckPrimitiveType(in_cnode, prim::kPrimTranspose)) { in HandleGraphMultiNode()
557 if (!CheckPrimitiveType(return_node->input(i), prim::kPrimTranspose)) { in SetSubGraphOutput()
756 …if (!CheckPrimitiveType(cnode, prim::kPrimTranspose) || GetTransposePerm(cnode, &perm) != lite::RE… in DecreaseTransposeForMultiOp()
|
| D | infershape_pass.cc | 56 if (CheckPrimitiveType(real_cnode, prim::kPrimTranspose)) { in GetCNodeCertainInputFormat()
308 if (!opt::CheckPrimitiveType(return_node->input(i), prim::kPrimTranspose)) { in SetSubGraphOutput()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/format/ |
| D | delete_redundant_transpose.cc | 56 if (!CheckPrimitiveType(node, prim::kPrimTranspose)) { in DeleteNot4DTranspose()
117 if (!CheckPrimitiveType(cnode, prim::kPrimTranspose) || in TransTransFusion()
118 !CheckPrimitiveType(cnode->input(1), prim::kPrimTranspose)) { in TransTransFusion()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | transdata_split.cc | 81 … prim::kPrimTranspose->name(), std::vector<int64_t>{3, 2, 0, 1}); in DoSplit()
87 … false, prim::kPrimTranspose->name(), std::vector<int64_t>{2, 3, 1, 0}); in DoSplit()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | parameter_and_transop_fusion.cc | 41 if (op_name == prim::kPrimCast->name() || op_name == prim::kPrimTranspose->name() || in ParamTransRoad()
97 if (node_name == prim::kPrimCast->name() || node_name == prim::kPrimTranspose->name() || in Run()
|
| D | transpose_transdata_fusion.cc | 27 VectorRef transpose({prim::kPrimTranspose, input_varptr_}); in DefinePattern()
|
| D | reshape_transpose_fusion.cc | 46 return VectorRef({prim::kPrimTranspose, reshape}); in DefinePattern()
|
| D | transpose_reshape_fusion.cc | 43 VectorRef transpose({prim::kPrimTranspose, input_varptr_}); in DefinePattern()
|
| D | transposed_update_fusion.cc | 68 return VectorRef({prim::kPrimTranspose, X}); in DefinePattern()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | transpose_eliminate.h | 37 AnfVisitor::Match(prim::kPrimTranspose, {IsNode, IsVNode})(node); in operator()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/ |
| D | insert_transpose_for_dynamic_gru_v2.cc | 60 prim::kPrimTranspose->name(), std::vector<int64_t>{2, 3, 1, 0}); in Insert()
|
| D | insert_transpose_for_basiclstm_op.cc | 42 auto prim = std::make_shared<Primitive>(prim::kPrimTranspose->name()); in Insert()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | primitive_infer_map.cc | 45 const auto kTranspose = prim::kPrimTranspose->name(); in GetDependsFormMap()
245 {prim::kPrimTranspose, {InferImplTranspose, nullptr, true}}, in GetPrimitiveToBackendEvalImplMap()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | transpose.cc | 99 REGISTER_PRIMITIVE_EVAL_IMPL(Transpose, prim::kPrimTranspose, TransposeInfer, nullptr, true);
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/ |
| D | const_input_to_attr_registry.cc | 47 Register(prim::kPrimTranspose->name(), {1}); in ConstInputToAttrInfoRegistry()
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | transformation_ops_declare.cc | 48 REG_ADPT_DESC(TransposeD, prim::kPrimTranspose->name(), ADPT_DESC(TransposeD))
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/ |
| D | tf_bidirection_gru_cf_fusion.cc | 170 auto fw_out_trans = VectorRef({GetPrim(prim::kPrimTranspose), tensor_array_gather, concat2}); in DefineBidirectionRnnPattern()
185 auto bw_transpose = VectorRef({GetPrim(prim::kPrimTranspose), bw_reverse_in, bw_concat}); in DefinePattern()
|
| D | inputs_adjust.cc | 113 if (opt::CheckPrimitiveType(node, prim::kPrimTranspose)) { in Run()
|