Huawei_Technology
/
mindspore
Not watched
1
0
Fork 0
Code
Releases 13 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
Browse Source

mod_matmul

tags/v1.1.0
wanyiming 5 years ago
parent
ff85533155
commit
a3458a2fdf
1 changed files with 0 additions and 1 deletions
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +0
    -1
      mindspore/nn/layer/math.py

+ 0
- 1
mindspore/nn/layer/math.py View File

@@ -354,7 +354,6 @@ def check_col_row_equal(x1_shape, x2_shape, transpose_x1, transpose_x2):
+ f'the row of matrix dimensions of x2, but got {x1_col} and {x2_row}.')


@constexpr
def matmul_op_select(x1_shape, x2_shape, transpose_x1, transpose_x2):
"""select matmul op"""
x1_dim, x2_dim = len(x1_shape), len(x2_shape)


Write Preview
Loading…
Cancel
Save