| /third_party/mindspore/mindspore/lite/micro/coder/wrapper/fp32/ |
| D | matmul_fp32_wrapper.c | 36 if (params_->b_transpose_) { in InitMatrixB()
51 if (params_->b_transpose_) { in InitMatrixB()
57 if (params_->b_transpose_) { in InitMatrixB()
|
| /third_party/mindspore/mindspore/lite/test/ut/nnacl/infer/ |
| D | matmul_infer_test.cc | 41 parameter->b_transpose_ = true; in TEST_F()
74 parameter->b_transpose_ = false; in TEST_F()
108 parameter->b_transpose_ = true; in TEST_F()
142 parameter->b_transpose_ = true; in TEST_F()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | matmul_fp32_base.cc | 128 if (params_->b_transpose_) { in CalBroadCastBiasDataElements()
195 if (params_->b_transpose_) { in InitMatrixB()
219 if (params_->b_transpose_) { in InitMatrixB()
307 matrix_b_pack_fun_ = params_->b_transpose_ ? RowMajor2Col16Major : RowMajor2Row16Major; in init_global_variable()
312 matrix_b_pack_fun_ = params_->b_transpose_ ? RowMajor2Col4Major : RowMajor2Row4Major; in init_global_variable()
317 matrix_b_pack_fun_ = params_->b_transpose_ ? RowMajor2Col8Major : RowMajor2Row8Major; in init_global_variable()
322 matrix_b_pack_fun_ = params_->b_transpose_ ? RowMajor2Col8Major : RowMajor2Row8Major; in init_global_variable()
|
| D | matmul_fp32.cc | 48 params_->col_ = params_->b_transpose_ ? b_shape[b_shape.size() - 2] : b_shape[b_shape.size() - 1]; in InitShapeB()
49 …params_->deep_ = params_->b_transpose_ ? b_shape[b_shape.size() - 1] : b_shape[b_shape.size() - 2]; in InitShapeB()
|
| D | fullconnection_fp32.cc | 49 params_->b_transpose_ = true; in Init()
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/ |
| D | matmul_fp32_tests.cc | 120 matmul_param->b_transpose_ = false; in TEST_F()
152 matmul_param->b_transpose_ = false; in TEST_F()
186 matmul_param->b_transpose_ = false; in TEST_F()
278 matmul_param->b_transpose_ = true; in TEST_F()
309 matmul_param->b_transpose_ = true; in TEST_F()
|
| D | fullconnection_fp32_tests.cc | 67 matmal_param->b_transpose_ = true; in FcTestInit1()
138 matmal_param->b_transpose_ = true; in FcTestInit2()
201 matmal_param->b_transpose_ = true; in FcTestInit3()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/ |
| D | matmul_cpu_kernel.cc | 98 if (param_.b_transpose_) { in InitMatrixB()
118 if (param_.b_transpose_) { in InitMatrixB()
124 if (param_.b_transpose_) { in InitMatrixB()
130 if (param_.b_transpose_) { in InitMatrixB()
144 param_.b_transpose_ = trans_b; in InitArmKernel()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | matmul_fp16.cc | 54 params_->col_ = params_->b_transpose_ ? b_shape[b_shape.size() - 2] : b_shape[b_shape.size() - 1]; in InitBShape()
56 …params_->deep_ = params_->b_transpose_ ? b_shape[b_shape.size() - 1] : b_shape[b_shape.size() - 2]; in InitBShape()
|
| D | fullconnection_fp16.cc | 60 params_->b_transpose_ = true; in Init()
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/ |
| D | matmul_fp32_coder.cc | 55 …params_->col_ = params_->b_transpose_ ? b_shape.at(b_shape_size - DIMENSION_2D) : b_shape.at(b_sha… in InitShapeB()
56 …params_->deep_ = params_->b_transpose_ ? b_shape.at(b_shape_size - 1) : b_shape.at(b_shape_size - … in InitShapeB()
|
| D | full_connection_fp32_coder.cc | 62 params_->b_transpose_ = true; in Init()
|
| /third_party/mindspore/mindspore/lite/src/ops/populate/ |
| D | matmul_populate.cc | 35 param->b_transpose_ = value->transpose_b(); in PopulateMatMulParameter()
|
| D | full_connection_populate.cc | 39 param->b_transpose_ = true; in PopulateFullconnectionParameter()
|
| D | affine_populate.cc | 55 matmul_param->b_transpose_ = value->transpose_b(); in PopulateAffineParameter()
|
| /third_party/mindspore/mindspore/lite/src/ops/populate/v0/ |
| D | matmul_populate_v0.cc | 37 matmul_param->b_transpose_ = value->transposeB(); in PopulateMatMulParameter()
|
| D | full_connection_populate_v0.cc | 39 matmul_param->b_transpose_ = true; in PopulateFullconnectionParameter()
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/ |
| D | fullconnection_int8_coder.cc | 50 param_->b_transpose_ = true; in Prepare()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | fullconnection_int8.cc | 32 param_->b_transpose_ = true; in Init()
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/ |
| D | matmul_int8_tests.cc | 120 matmul_param->b_transpose_ = true; in TEST_F()
238 matmul_param->b_transpose_ = false; in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/ |
| D | matmul_parameter.h | 59 bool b_transpose_; /* true : col-major */ member
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | matmul_infer.c | 38 if (param->b_transpose_) { in CheckMatmulInputShape()
|
| D | affine_infer.c | 30 if (matmul_param->b_transpose_) { in MatmulInfer()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/ |
| D | fullconnection.cc | 68 if (!param->b_transpose_) { in CheckSpecs()
93 transposeB = param->b_transpose_; in Prepare()
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/ |
| D | matmul_tests.cc | 28 param->b_transpose_ = b_transpose; in CreateParameter()
|