| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/ |
| D | matmul_fp32_tests.cc | 74 …const std::vector<int> &a_shape, const std::vector<int> &b_shape, const std::vector<int> &c_shape)… in MMTestInit()
93 float *bias_ptr, const std::vector<int> &a_shape, const std::vector<int> &b_shape, in MMTestInit2()
133 std::vector<int> b_shape = {8, 3}; in TEST_F() local
170 std::vector<int> b_shape = {8, 3}; in TEST_F() local
227 std::vector<int> b_shape = {12, 36}; in TEST_F() local
301 std::vector<int> b_shape = {1, 3, 8}; in TEST_F() local
355 std::vector<int> b_shape = {3, 3, 8}; in TEST_F() local
|
| D | roi_pooling_fp32_tests.cc | 29 … float *b_ptr, const std::vector<int> &a_shape, const std::vector<int> &b_shape, in ROIPoolingTestInit()
59 std::vector<int> b_shape = {2, 5}; in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/ |
| D | matmul.c | 41 int *b_shape = matmul->base_.in_[kWeightIndex]->shape_; in MatmulInitShapeB() local
70 int b_shape[MAX_SHAPE_SIZE] = {0}; in MatmulInitBroadcastParams() local
|
| D | fullconnection.c | 34 int *b_shape = self->in_[SECOND_INPUT]->shape_; in FullConnectionPrepare() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp16/ |
| D | fullconnection_fp16.cc | 35 auto b_shape = in_tensors_.at(1)->shape(); in InitBShape() local
|
| D | matmul_fp16.cc | 47 auto b_shape = in_tensors_[1]->shape(); in InitBShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | betainc.cc | 50 auto b_shape = ctx.Input(1)->GetTensorShape(); in Compute() local
123 auto b_shape = ctx.Input(1)->GetTensorShape()->GetDimSizes(); in BetaincCompute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/ |
| D | matmul_infer.c | 23 int CheckMatmulInputShape(int *a_shape, size_t a_shape_size, int *b_shape, size_t b_shape_size, con… in CheckMatmulInputShape()
74 int b_shape[MAX_SHAPE_SIZE] = {0}; in SetShape() local
|
| D | affine_infer.c | 72 int b_shape[MAX_SHAPE_SIZE] = {0}; in AffineInferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp32/ |
| D | full_connection_fp32_coder.cc | 59 std::vector<int> b_shape = filter_tensor_->shape(); in Init() local
|
| D | matmul_fp32_coder.cc | 44 std::vector<int> b_shape = filter_tensor_->shape(); in InitShapeB() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp16/ |
| D | matmul_dynamic_fp16_coder.cc | 35 std::vector<int> b_shape = filter_tensor_->shape(); in InitBShape() local
|
| D | matmul_fp16_coder.cc | 40 std::vector<int> b_shape = filter_tensor_->shape(); in InitBShape() local
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/dynamic_shape/grad/ |
| D | test_sparse_add.py | 28 def construct(self, a_indices, a_values, a_shape, b_indices, b_values, b_shape, thresh): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/eigen/ |
| D | solve_triangular_cpu_kernel.cc | 57 auto b_shape = LongVecToSizeVec(inputs.at(kIndexB)->GetShapeVector()); in Resize() local
116 auto b_shape = LongVecToSizeVec(inputs.at(kIndexB)->GetShapeVector()); in SolveTriangularCheck() local
|
| D | lu_solve_cpu_kernel.cc | 43 auto b_shape = Convert2SizeTClipNeg(inputs[kLUbIndex]->GetShapeVector()); in Resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/ |
| D | betainc.cc | 37 const auto &b_shape = b_shape_base->GetShapeVector(); in InferShape() local
|
| D | solve_triangular.cc | 33 auto b_shape = input_args[kInputIndex1]->GetShape()->GetShapeVector(); in InferShape() local
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/ |
| D | test_sparse_matrix_add_op.py | 30 a_values, b_shape, b_batch_pointer, b_indptr, b_indices, argument
|
| D | test_sparse_add_ops.py | 31 def construct(self, a_indices, a_values, a_shape, b_indices, b_values, b_shape, thresh): argument
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/ |
| D | test_ops_betainc.cc | 34 ShapeVector b_shape; member
|
| D | test_ops_solve_triangular.cc | 38 ShapeVector b_shape; member
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/ |
| D | dense_grad.cc | 72 ShapeVector b_shape = {w_shape[kDenseGradIndex0]}; in InferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | fused_matmul_bias_add.cc | 74 auto b_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(b_shape_ptr)[kShape]; in InferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/ |
| D | matmul_fp32_base.cc | 436 auto b_shape = in_tensors_.at(1)->shape(); in FullConnectionPrepare() local
479 auto b_shape = in_tensors_[kWeightIndex]->shape(); in InitShapeB() local
588 std::vector<int> b_shape = b_shape_const; in InitBroadcastParams() local
|