| /third_party/mindspore/mindspore/lite/test/ut/nnacl/infer/ |
| D | matmul_infer_test.cc | 40 parameter->a_transpose_ = false; in TEST_F()
73 parameter->a_transpose_ = false; in TEST_F()
107 parameter->a_transpose_ = false; in TEST_F()
141 parameter->a_transpose_ = false; in TEST_F()
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/ |
| D | matmul_fp32_tests.cc | 119 matmul_param->a_transpose_ = false; in TEST_F()
151 matmul_param->a_transpose_ = false; in TEST_F()
185 matmul_param->a_transpose_ = false; in TEST_F()
277 matmul_param->a_transpose_ = false; in TEST_F()
308 matmul_param->a_transpose_ = false; in TEST_F()
|
| D | fullconnection_fp32_tests.cc | 68 matmal_param->a_transpose_ = false; in FcTestInit1()
139 matmal_param->a_transpose_ = false; in FcTestInit2()
202 matmal_param->a_transpose_ = false; in FcTestInit3()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | matmul_fp32.cc | 36 params_->row_ = params_->a_transpose_ ? a_shape[a_shape.size() - 1] : a_shape[a_shape.size() - 2]; in InitShapeA()
37 …params_->deep_ = params_->a_transpose_ ? a_shape[a_shape.size() - 2] : a_shape[a_shape.size() - 1]; in InitShapeA()
|
| D | matmul_fp32_base.cc | 180 if (params_->a_transpose_) { in InitMatrixA()
306 matrix_a_pack_fun_ = params_->a_transpose_ ? RowMajor2Row6Major : RowMajor2Col6Major; in init_global_variable()
311 matrix_a_pack_fun_ = params_->a_transpose_ ? RowMajor2Row12Major : RowMajor2Col12Major; in init_global_variable()
316 matrix_a_pack_fun_ = params_->a_transpose_ ? RowMajor2Row4Major : RowMajor2Col4Major; in init_global_variable()
321 matrix_a_pack_fun_ = params_->a_transpose_ ? RowMajor2Row12Major : RowMajor2Col12Major; in init_global_variable()
|
| D | fullconnection_fp32.cc | 48 params_->a_transpose_ = false; in Init()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | matmul_fp16.cc | 38 params_->row_ = params_->a_transpose_ ? a_shape[a_shape.size() - 1] : a_shape[a_shape.size() - 2]; in InitAShape()
39 …params_->deep_ = params_->a_transpose_ ? a_shape[a_shape.size() - 2] : a_shape[a_shape.size() - 1]; in InitAShape()
|
| D | fullconnection_fp16.cc | 59 params_->a_transpose_ = false; in Init()
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/ |
| D | matmul_fp32_coder.cc | 38 …params_->row_ = params_->a_transpose_ ? a_shape.at(a_shape_size - 1) : a_shape.at(a_shape_size - D… in InitShapeA()
39 …params_->deep_ = params_->a_transpose_ ? a_shape.at(a_shape_size - DIMENSION_2D) : a_shape.at(a_sh… in InitShapeA()
|
| D | full_connection_fp32_coder.cc | 61 params_->a_transpose_ = false; in Init()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/ |
| D | matmul_cpu_kernel.cc | 69 if (param_.a_transpose_) { in InitMatrixA()
75 if (param_.a_transpose_) { in InitMatrixA()
81 if (param_.a_transpose_) { in InitMatrixA()
143 param_.a_transpose_ = trans_a; in InitArmKernel()
|
| /third_party/mindspore/mindspore/lite/src/ops/populate/ |
| D | matmul_populate.cc | 36 param->a_transpose_ = value->transpose_a(); in PopulateMatMulParameter()
|
| D | full_connection_populate.cc | 40 param->a_transpose_ = false; in PopulateFullconnectionParameter()
|
| D | affine_populate.cc | 56 matmul_param->a_transpose_ = value->transpose_a(); in PopulateAffineParameter()
|
| /third_party/mindspore/mindspore/lite/src/ops/populate/v0/ |
| D | matmul_populate_v0.cc | 38 matmul_param->a_transpose_ = value->transposeA(); in PopulateMatMulParameter()
|
| D | full_connection_populate_v0.cc | 40 matmul_param->a_transpose_ = false; in PopulateFullconnectionParameter()
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/ |
| D | fullconnection_int8_coder.cc | 49 param_->a_transpose_ = false; in Prepare()
|
| D | matmul_int8_coder.cc | 48 …param_->deep_ = param_->a_transpose_ ? x_shape[x_shape.size() - kBiasIndex] : x_shape[x_shape.size… in ReSize()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | matmul_int8.cc | 59 param_->deep_ = param_->a_transpose_ ? x_shape[x_shape.size() - 2] : x_shape[x_shape.size() - 1]; in ReSize()
|
| D | fullconnection_int8.cc | 31 param_->a_transpose_ = false; in Init()
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/ |
| D | matmul_int8_tests.cc | 119 matmul_param->a_transpose_ = false; in TEST_F()
237 matmul_param->a_transpose_ = false; in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/ |
| D | matmul_parameter.h | 58 bool a_transpose_; /* false : row-major */ member
|
| /third_party/mindspore/mindspore/lite/micro/coder/wrapper/fp32/ |
| D | matmul_fp32_wrapper.c | 26 if (params_->a_transpose_) { in InitMatrixA()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | matmul_infer.c | 32 if (param->a_transpose_) { in CheckMatmulInputShape()
|
| D | affine_infer.c | 24 if (matmul_param->a_transpose_) { in MatmulInfer()
|