Searched refs:kPrimBatchMatMul (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | batch_matmul.h | 32 …BatchMatmul() : PrimitiveC(prim::kPrimBatchMatMul->name()) { InitIOName({"x1", "x2"}, {"output"});… in BatchMatmul()
|
| D | batch_matmul.cc | 157 REGISTER_PRIMITIVE_EVAL_IMPL(BatchMatmul, prim::kPrimBatchMatMul, BatchMatmulInfer, nullptr, true);
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/ |
| D | batchmatmul_fusedmuladd_fusion_pass.cc | 32 …(batch_matmul->isa<CNode>() && AnfAlgo::CheckPrimitiveType(batch_matmul, prim::kPrimBatchMatMul)) { in MatchBatchMatmulFusedMulAdd()
|
| D | matmul_confusiontranspose_fusion_pass.cc | 34 AnfAlgo::CheckPrimitiveType(matmul, prim::kPrimBatchMatMul))) { in MatchMatmulConfusionTranpose()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | dynamic_gru_v2_grad_fission.cc | 291 …dePtr> matmul_inputs = {NewValueNode(std::make_shared<Primitive>(prim::kPrimBatchMatMul->name()))}; in AddDwhMatmulNode()
365 …odePtr> matmul_inputs = {NewValueNode(std::make_shared<Primitive>(prim::kPrimBatchMatMul->name())), in CreateDwxBatchMatMul()
383 …odePtr> matmul_inputs = {NewValueNode(std::make_shared<Primitive>(prim::kPrimBatchMatMul->name())), in CreateDxtBatchMatMul()
|
| D | dynamic_rnn_grad_fission_v2.cc | 418 …odePtr> matmul_inputs = {NewValueNode(std::make_shared<Primitive>(prim::kPrimBatchMatMul->name())), in CreateBatchMatMul()
437 …odePtr> matmul_inputs = {NewValueNode(std::make_shared<Primitive>(prim::kPrimBatchMatMul->name())), in CreateBatchMatMul2()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | graph_kernel_expander.cc | 67 {kGPUDevice, OpLevel_1, prim::kPrimBatchMatMul}, in GetExpandOps()
|
| D | graph_kernel_cluster.cc | 60 {kAscendDevice, OpLevel_1, prim::kPrimBatchMatMul}, in GetClusterableOpList()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | primitive_infer_map.cc | 103 {prim::kPrimBatchMatMul, {InferImplBatchMatMul, nullptr, true}}, in GetPrimitiveToEvalImplMap()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 468 inline const PrimitivePtr kPrimBatchMatMul = std::make_shared<Primitive>("BatchMatMul"); variable
|
| /third_party/mindspore/mindspore/ccsrc/transform/express_ir/ |
| D | onnx_exporter.cc | 710 if (cnode->IsApply(prim::kPrimBatchMatMul)) { in ExportNodes()
|