Searched refs:multiples_ (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | tile_infer.c | 31 multiples_size_tmp[param->dims_[i]] = param->multiples_[i]; in TileParamCaffe2Tflite()
34 param->multiples_[i] = multiples_size_tmp[i]; in TileParamCaffe2Tflite()
76 param->multiples_[i] = input1_data[i]; in TileInferShape()
101 if (input->shape_[dims[i]] != 0 && param->multiples_[i] > INT_MAX / input->shape_[dims[i]]) { in TileInferShape()
104 MS_CHECK_FALSE(INT_MUL_OVERFLOW(input->shape_[dims[i]], (param->multiples_[i])), NNACL_ERR); in TileInferShape()
105 out_shape[dims[i]] = input->shape_[dims[i]] * (param->multiples_[i]); in TileInferShape()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | tile_cpu_kernel.cc | 32 for (size_t i = 0; i < multiples_.size(); i++) { in TileMultipleCompute()
33 tile_parameter_.multiples_[i] = multiples_[i]; in TileMultipleCompute()
34 if (tile_parameter_.multiples_[i] > 1) { in TileMultipleCompute()
36 multiple = tile_parameter_.multiples_[i]; in TileMultipleCompute()
60 (void)std::transform(multiples_me.begin(), multiples_me.end(), std::back_inserter(multiples_), in TileTensorParamrInit()
63 size_t ones = multiples_.size() - x_shape_.size(); in TileTensorParamrInit()
|
| D | tile_cpu_kernel.h | 49 std::vector<int> multiples_; variable
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/ |
| D | tile_fp32_tests.cc | 42 parameter.multiples_[0] = 2; in TEST_F()
43 parameter.multiples_[1] = 3; in TEST_F()
88 parameter.multiples_[0] = 2; in TEST_F()
89 parameter.multiples_[1] = 1; in TEST_F()
136 parameter.multiples_[0] = 1; in TEST_F()
137 parameter.multiples_[1] = 2; in TEST_F()
|
| /third_party/mindspore/mindspore/lite/test/ut/nnacl/infer/ |
| D | tile_infer_test.cc | 38 parameter->multiples_[0] = 4; in TEST_F()
39 parameter->multiples_[1] = 5; in TEST_F()
71 parameter->multiples_[0] = 4; in TEST_F()
72 parameter->multiples_[1] = 5; in TEST_F()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/ |
| D | tile_base.cc | 57 tile_parameter_->multiples_[i] = input1_addr[i]; in ReSize()
100 if (tile_parameter_->multiples_[i] > 1) { in FillOneDimTileParam()
102 multiple = tile_parameter_->multiples_[i]; in FillOneDimTileParam()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/ |
| D | tile_base.c | 32 …DoCopyData(input_data, output_data, src_dim_size, parameter->data_size_, parameter->multiples_[dim… in DoTileOneDimension()
36 for (int j = 0; j < parameter->multiples_[dim]; ++j) { in DoTileOneDimension()
|
| D | tile_base.h | 25 int multiples_[MAX_TILE_DIM_SIZE]; member
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/serializers/nnacl_serializer/ |
| D | nnacl_fp32_serializer.cc | 98 …aseStruct("TileParameter", name, tile_parameter.op_parameter_, ToString(tile_parameter.multiples_), in CodeStruct()
|