Browse Source

!26217 Ge Adapter BatchMaMul

Merge pull request !26217 from weiyang/ge_adpator_batchmatmul
tags/v1.6.0
i-robot Gitee 4 years ago
parent
commit
50f4f96c77
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/transform/graph_ir/op_declare/matrix_calculation_ops_declare.cc

+ 1
- 1
mindspore/ccsrc/transform/graph_ir/op_declare/matrix_calculation_ops_declare.cc View File

@@ -132,7 +132,7 @@ INPUT_MAP(BatchMatMulV2) = {{1, INPUT_DESC(x1)}, {2, INPUT_DESC(x2)}};
ATTR_MAP(BatchMatMulV2) = {{"transpose_x1", ATTR_DESC(adj_x1, AnyTraits<bool>())},
{"transpose_x2", ATTR_DESC(adj_x2, AnyTraits<bool>())}};
OUTPUT_MAP(BatchMatMulV2) = {{0, OUTPUT_DESC(y)}};
REG_ADPT_DESC(BatchMatMul, kNameBatchMatMul, ADPT_DESC(BatchMatMulV2))
REG_ADPT_DESC(BatchMatMul, kNameBatchMatMul, ADPT_DESC(BatchMatMul))
REG_ADPT_DESC(BatchMatMulV2, kNameBatchMatMulV2, ADPT_DESC(BatchMatMulV2))

// L2Loss


Loading…
Cancel
Save