Searched defs:is_reduce (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/ |
| D | tf_bidirection_gru_fusion.cc | 135 auto is_reduce = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimReduceFusion)); in DefineFowardPattern() local
201 auto is_reduce = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimReduceFusion)); in DefinebackwardPattern() local
|
| D | decoder_layer_fusion.cc | 141 …auto is_reduce = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimReduceFusion), "redu… in DefineLayerNorm() local
|
| D | encoder_layer_fusion.cc | 147 …auto is_reduce = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimReduceFusion), "redu… in DefineLayerNorm() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/_extends/graph_kernel/model/ |
| D | model.py | 266 def is_reduce(cls, op): member in PrimLib
|