Home
last modified time | relevance | path

Searched refs:DimensionSize (Results 1 – 24 of 24) sorted by relevance

/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Dmul_int8.cc113 tile_para->in_shape0_[i] = in_tensors_.at(0)->DimensionSize(i); in ReSize()
114 tile_para->in_shape1_[i] = in_tensors_.at(1)->DimensionSize(i); in ReSize()
115 tile_para->out_shape_[i] = out_tensors_.at(0)->DimensionSize(i); in ReSize()
124 tile_para->in_shape0_[i] = in_tensors_.at(0)->DimensionSize(j++); in ReSize()
126 tile_para->in_shape1_[i] = in_tensors_.at(1)->DimensionSize(i); in ReSize()
127 tile_para->out_shape_[i] = out_tensors_.at(0)->DimensionSize(i); in ReSize()
133 tile_para->in_shape0_[i] = in_tensors_.at(0)->DimensionSize(i); in ReSize()
137 tile_para->in_shape1_[i] = in_tensors_.at(1)->DimensionSize(j++); in ReSize()
139 tile_para->out_shape_[i] = out_tensors_.at(0)->DimensionSize(i); in ReSize()
Ddiv_int8.cc113 tile_para.in_shape0_[i] = in_tensors_.at(0)->DimensionSize(i); in Run()
114 tile_para.in_shape1_[i] = in_tensors_.at(1)->DimensionSize(i); in Run()
115 tile_para.out_shape_[i] = out_tensors_.at(0)->DimensionSize(i); in Run()
Dscale_int8.cc149 tile_para->in_shape0_[i] = in_tensors_.at(0)->DimensionSize(i); in InitParameter()
150 tile_para->in_shape1_[i] = in_tensors_.at(1)->DimensionSize(i); in InitParameter()
151 tile_para->out_shape_[i] = out_tensors_.at(0)->DimensionSize(i); in InitParameter()
158 tile_para->in_shape0_[i] = in_tensors_.at(0)->DimensionSize(i); in InitParameter()
164 tile_para->out_shape_[i] = out_tensors_.at(0)->DimensionSize(i); in InitParameter()
Dsub_int8.cc139 tile_para.in_shape0_[i] = in_tensors_.at(0)->DimensionSize(i); in Run()
140 tile_para.in_shape1_[i] = in_tensors_.at(1)->DimensionSize(i); in Run()
141 tile_para.out_shape_[i] = out_tensors_.at(0)->DimensionSize(i); in Run()
Dconcat_int8.cc97 before_axis_size *= out_tensors_.at(kOutputIndex)->DimensionSize(i); in ReSize()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dreverse_sequence_fp32.cc72 para->input_shape0_[i] = input0->DimensionSize(i); in ReSize()
73 para->output_shape_[i] = output->DimensionSize(i); in ReSize()
80 …para->outer_stride_ = input0->DimensionSize(less_axis) * CalcCountAfterAxis(input0->shape(), less_… in ReSize()
84 para->inner_count_ *= input0->DimensionSize(i); in ReSize()
87 …para->inner_stride_ = input0->DimensionSize(greater_axis) * CalcCountAfterAxis(input0->shape(), gr… in ReSize()
Dunstack_fp32.cc53 para->pre_dims_ *= input->DimensionSize(i); in ReSize()
55 para->after_dims_ *= input->DimensionSize(i); in ReSize()
57 para->axis_dim_ = input->DimensionSize(i); in ReSize()
Ddeconvolution_winograd_fp32.cc319 …if (bias_tensor->shape().size() == 1 && bias_tensor->DimensionSize(0) == conv_param_->output_chann… in InitDataParam()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/cmsis-nn/int8/
Dconv2d_int8_coder.cc99 MS_CHECK_TRUE(input_tensor_->Channel() == filter_tensor_->DimensionSize(kNHWC_C), in SetParameters()
101 MS_CHECK_TRUE(output_tensor_->Channel() == filter_tensor_->DimensionSize(kNHWC_N), in SetParameters()
109 kernel_x_ = filter_tensor_->DimensionSize(kNHWC_W); in SetParameters()
110 kernel_y_ = filter_tensor_->DimensionSize(kNHWC_H); in SetParameters()
125 output_x_ = output_tensor_->DimensionSize(kNHWC_W); in SetParameters()
126 output_y_ = output_tensor_->DimensionSize(kNHWC_H); in SetParameters()
Dsoftmax_int8_coder.cc56 num_rows_ *= (i == trailing_dim) ? 1 : input_tensor_->DimensionSize(i); in Prepare()
59 …MS_CHECK_TRUE(input_tensor_->DimensionSize(trailing_dim) == output_tensor_->DimensionSize(trailing… in Prepare()
61 row_size_ = input_tensor_->DimensionSize(trailing_dim); in Prepare()
Dfullconnection_int8_coder.cc72 col_dim_ = filter_tensor_->DimensionSize(filter_tensor_->shape().size() - 1); in SetParameters()
73 row_dim_ = output_tensor_->DimensionSize(1); in SetParameters()
Dpooling_int8_coder.cc80 dim_dst_height_ = output_tensor_->DimensionSize(1); in SetParameters()
81 dim_dst_width_ = output_tensor_->DimensionSize(2); in SetParameters()
/third_party/mindspore/tests/ut/cpp/ir/
Dmeta_tensor_test.cc52 ASSERT_EQ(2, meta_tensor.DimensionSize(0)); in TEST_F()
53 ASSERT_EQ(3, meta_tensor.DimensionSize(1)); in TEST_F()
54 ASSERT_EQ(-1, meta_tensor.DimensionSize(2)); in TEST_F()
71 ASSERT_EQ(5, meta_tensor_.DimensionSize(0)); in TEST_F()
72 ASSERT_EQ(6, meta_tensor_.DimensionSize(1)); in TEST_F()
73 ASSERT_EQ(7, meta_tensor_.DimensionSize(2)); in TEST_F()
179 ASSERT_EQ(2, tuple_tensor.DimensionSize(0)); in TEST_F()
180 ASSERT_EQ(3, tuple_tensor.DimensionSize(1)); in TEST_F()
181 ASSERT_EQ(4, tuple_tensor.DimensionSize(2)); in TEST_F()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/
Ddiv_int8_coder.cc62 tile_para.in_shape0_[i] = input0->DimensionSize(i); in DoCode()
63 tile_para.in_shape1_[i] = input1->DimensionSize(i); in DoCode()
64 tile_para.out_shape_[i] = output_tensor_->DimensionSize(i); in DoCode()
Dsub_int8_coder.cc84 tile_para.in_shape0_[i] = input0->DimensionSize(i); in DoCode()
85 tile_para.in_shape1_[i] = input1->DimensionSize(i); in DoCode()
86 tile_para.out_shape_[i] = output_tensor_->DimensionSize(i); in DoCode()
Dconcat_int8_coder.cc73 before_axis_size *= output_tensor_->DimensionSize(i); in Prepare()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/string/
Dlsh_projection.cc65 param_->hash_shape_[0] = input0_tensor->DimensionSize(0); in Run()
66 param_->hash_shape_[1] = input0_tensor->DimensionSize(1); in Run()
/third_party/mindspore/mindspore/core/ir/
Dmeta_tensor.cc64 int64_t MetaTensor::DimensionSize(const size_t index) const { in DimensionSize() function in mindspore::tensor::MetaTensor
Dmeta_tensor.h147 int64_t DimensionSize(size_t index) const;
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/
Dslice_base.cc61 param_->shape_[i] = in_tensor->DimensionSize(i); in ReSize()
/third_party/mindspore/mindspore/lite/src/
Dtensor.h97 int DimensionSize(size_t index) const;
Dtensor.cc237 int Tensor::DimensionSize(const size_t index) const { in DimensionSize() function in mindspore::lite::Tensor
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Ddeconvolution_winograd_fp16.cc358 …if (bias_tensor->shape().size() == 1 && bias_tensor->DimensionSize(0) == conv_param_->output_chann… in InitDataParam()
/third_party/mindspore/
DRELEASE.md1293 ###### Delete `DimensionSize()` in ms_tensor.h.([!13515](https://gitee.com/mindspore/mindspore/pull…
1295 …named `DimensionSize` is fuinctionally overlapped with the interface named `shape`. For the simpli…
1299 <td style="text-align:center"> DimensionSize() </td>
1310 virtual int DimensionSize(size_t index) const = 0;