Home
last modified time | relevance | path

Searched defs:bias_add (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dmatmul_biasadd_fusion.cc58 …VectorRef bias_add = VectorRef({prim::kPrimBiasAdd, VectorRef({prim::kPrimMatMul, x_, load_w}), lo… in DefinePattern() local
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dmulti_head_attention_fusion.cc130 auto bias_add = VectorRef({is_bias_add, reshape2, bias}); in DefineDensePattern() local
/third_party/mindspore/mindspore/lite/mindir/src/
Dmindir_nnrt_lite_graph_to_model.cc333 BiasAdd bias_add{}; in ConvertBiasAdd() local