Searched refs:kPrimBiasAdd (Results 1 – 13 of 13) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | bias_add.h | 36 BiasAdd() : PrimitiveC(prim::kPrimBiasAdd->name()) { InitIOName({"x", "b"}, {"output"}); } in BiasAdd()
|
| D | bias_add.cc | 112 REGISTER_PRIMITIVE_EVAL_IMPL(BiasAdd, prim::kPrimBiasAdd, BiasAddInfer, nullptr, true);
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | matmul_biasadd_fusion.cc | 28 VectorRef pattern({prim::kPrimBiasAdd, matmul, x2_}); in DefinePattern()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | matmul_add_fusion.cc | 58 …!CheckPrimitiveType(node, prim::kPrimAddFusion) && !CheckPrimitiveType(node, prim::kPrimBiasAdd)) { in Run()
|
| D | transpose_fusion.cc | 96 auto is_bias_add = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimBiasAdd>); in DefineBiasAddPattern()
|
| D | conv_biasadd_fusion.cc | 39 …kPrimitiveType(anf_node, prim::kPrimAddFusion) || CheckPrimitiveType(anf_node, prim::kPrimBiasAdd); in IsAddNode()
|
| D | multi_head_attention_fusion.cc | 128 auto is_bias_add = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimBiasAdd)); in DefineDensePattern()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | matmul_biasadd_fusion.cc | 58 …VectorRef bias_add = VectorRef({prim::kPrimBiasAdd, VectorRef({prim::kPrimMatMul, x_, load_w}), lo… in DefinePattern()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/ |
| D | tf_bidirection_gru_cf_fusion.cc | 63 auto bias = VectorRef({GetPrim(prim::kPrimBiasAdd), matmul, bias_enter}); in DefineGruCellPattern()
73 auto bias1 = VectorRef({GetPrim(prim::kPrimBiasAdd), matmul1, bias1_enter}); in DefineGruCellPattern()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/cpu/ |
| D | insert_format_transform_op.cc | 173 const std::unordered_set<std::string> kChannelLastKernel = {prim::kPrimBiasAdd->name()};
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | graph_kernel_expander.cc | 68 {kGPUDevice, OpLevel_0, prim::kPrimBiasAdd}, in GetExpandOps()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | primitive_infer_map.cc | 160 {prim::kPrimBiasAdd, {InferImplBiasAdd, nullptr, true}}, in GetPrimitiveToEvalImplMap()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 328 inline const PrimitivePtr kPrimBiasAdd = std::make_shared<Primitive>(kBiasAdd); variable
|