Searched defs:is_transpose (Results 1 – 5 of 5) sorted by relevance
| /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() local
57 auto is_transpose = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTranspose>); in DefineActivationscalePattern() local
75 auto is_transpose = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTranspose>); in DefineActivationPattern() local
89 auto is_transpose = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTranspose>); in DefineBiasAddPattern() local
|
| D | multi_head_attention_fusion.cc | 38 auto is_transpose = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTranspose)); in DefineEmbedding() local
90 auto is_transpose = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTranspose)); in DefineMPWithMaskPattern() local
142 auto is_transpose = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTranspose)); in DefineProcessInputPattern() local
151 auto is_transpose = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTranspose)); in DefineProcessOutputPattern() local
|
| D | conv_pad_fusion.cc | 170 auto is_transpose = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTranspose>); in DefinePadTransposeConvPattern() local
|
| D | tflite_rel_pos_multi_head_attention_fusion.cc | 411 auto is_transpose = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTranspose)); in DefineProcessInputPattern() local
423 auto is_transpose = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTranspose)); in DefineProcessOutputPattern() local
|
| D | tf_bidirection_gru_fusion.cc | 181 auto is_transpose = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTranspose)); in DefineFowardPattern() local
203 auto is_transpose = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTranspose)); in DefinebackwardPattern() local
|