| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | cumsum_cpu_kernel.cc | 99 dims_[0] = 1; in Reshape()
100 dims_[1] = shape_[IntToSize(axis_)]; in Reshape()
101 dims_[2] = 1; in Reshape()
103 dims_[0] *= shape_[i]; in Reshape()
106 dims_[2] *= shape_[i]; in Reshape()
108 stride_ = dims_[1] * dims_[2]; in Reshape()
109 stride2_ = dims_[2]; in Reshape()
204 start = start / dims_[1]; in LaunchCumSum()
205 end = end / dims_[1]; in LaunchCumSum()
208 RightMove(input, output, dims_[0], dims_[1], dims_[2], stride_, stride2_, start, end); in LaunchCumSum()
[all …]
|
| D | random_choice_with_mask_cpu_kernel.cc | 60 void GetInputTotalCount(const std::vector<int32_t> &dims_, int32_t *input_total_count, const int32_… in GetInputTotalCount() argument
62 *input_total_count *= dims_[i]; in GetInputTotalCount()
66 void UpdateOutput(const std::vector<int32_t> &dims_, const int32_t &non_zero_num, const int32_t &co… in UpdateOutput() argument
68 …for (int32_t i = non_zero_num * SizeToInt(dims_.size()); i < count_ * SizeToInt(dims_.size()); i++… in UpdateOutput()
107 (void)dims_.emplace_back(input_i_shape[j]); in InitKernel()
110 input_dim_size = SizeToInt(dims_.size()); in InitKernel()
119 GetInputTotalCount(dims_, &input_total_count, input_dim_size); in InitInputOutputSize()
176 …ParseOutputCoordinate(dims_, output_length, input_dim_size, input_total_count, tmp_output, output); in Launch()
178 int32_t actual_output_length = count_ * SizeToInt(dims_.size()); in Launch()
189 UpdateOutput(dims_, non_zero_num, count_, output_length, mask_dim, output_coordinate, mask); in Launch()
|
| D | gathernd_cpu_kernel.cc | 54 (void)dims_.emplace_back(dim_of_indices); in InitKernel()
55 (void)dims_.emplace_back(dim_after_indices); in InitKernel()
56 (void)dims_.emplace_back(dim_indices_last); in InitKernel()
63 batch_indices_[dim_indices_last - 1] = dims_[1]; in InitKernel()
82 size_t output_dim0 = dims_[0]; in Launch()
83 size_t output_dim1 = dims_[1]; in Launch()
84 size_t indices_dim1 = dims_[2]; in Launch()
|
| D | random_choice_with_mask_cpu_kernel.h | 51 std::vector<int32_t> dims_; variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/ |
| D | cumprod_gpu_kernel.h | 53 …CumProd(input_addr, output_addr, ws_addr, dims_[0], dims_[1], dims_[2], stride_, stride2_, exclusi… in Launch()
97 dims_[0] = 1; in Reshape()
98 dims_[1] = shape_[IntToSize(axis_)]; in Reshape()
99 dims_[2] = 1; in Reshape()
101 dims_[0] *= shape_[i]; in Reshape()
104 dims_[2] *= shape_[i]; in Reshape()
106 stride_ = dims_[1] * dims_[2]; in Reshape()
107 stride2_ = dims_[2]; in Reshape()
117 size_t dims_[kMaxDimsSize] = {}; variable
|
| D | cumsum_gpu_kernel.h | 53 …CumSum(input_addr, output_addr, ws_addr, dims_[0], dims_[1], dims_[2], stride_, stride2_, exclusiv… in Launch()
97 dims_[0] = 1; in Reshape()
98 dims_[1] = shape_[IntToSize(axis_)]; in Reshape()
99 dims_[2] = 1; in Reshape()
101 dims_[0] *= shape_[i]; in Reshape()
104 dims_[2] *= shape_[i]; in Reshape()
106 stride_ = dims_[1] * dims_[2]; in Reshape()
107 stride2_ = dims_[2]; in Reshape()
117 size_t dims_[kMaxDimsSize] = {}; variable
|
| /third_party/mindspore/mindspore/lite/test/ut/nnacl/infer/ |
| D | fill_infer_test.cc | 40 parameter->dims_[0] = 1; in TEST_F()
41 parameter->dims_[1] = 2; in TEST_F()
42 parameter->dims_[2] = 3; in TEST_F()
43 parameter->dims_[3] = 4; in TEST_F()
69 parameter->dims_[0] = 4; in TEST_F()
70 parameter->dims_[1] = 2; in TEST_F()
71 parameter->dims_[2] = 3; in TEST_F()
96 parameter->dims_[0] = 4; in TEST_F()
97 parameter->dims_[1] = 2; in TEST_F()
121 parameter->dims_[0] = 4; in TEST_F()
|
| D | tile_infer_test.cc | 41 parameter->dims_[0] = 0; in TEST_F()
42 parameter->dims_[1] = 1; in TEST_F()
74 parameter->dims_[0] = 1; in TEST_F()
75 parameter->dims_[1] = 2; in TEST_F()
|
| D | unsqueeze_infer_test.cc | 37 parameter->dims_[0] = 0; in TEST_F()
63 parameter->dims_[0] = 1; in TEST_F()
116 parameter->dims_[0] = 1; in TEST_F()
148 parameter->dims_[0] = 1; in TEST_F()
180 parameter->dims_[0] = 3; in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | gather_grad_gpu_kernel.h | 47 GatherGrad(index_addr, grad_addr, output_addr, dims_[0], dims_[1], dims_[2], dims_[3], in Launch()
112 dims_[0] = dim_before_axis; in Reshape()
113 dims_[1] = dim_at_axis_index; in Reshape()
114 dims_[2] = dim_at_axis_output; in Reshape()
115 dims_[3] = dim_after_axis; in Reshape()
130 size_t dims_[4] = {}; variable
|
| D | gather_gpu_kernel.h | 47 Gather(input_addr, index_addr, output_addr, dims_[0], dims_[1], dims_[2], dims_[3], in Launch()
111 dims_[0] = dim_before_axis; in Reshape()
112 dims_[1] = dim_at_axis_input; in Reshape()
113 dims_[2] = dim_at_axis_output; in Reshape()
114 dims_[3] = dim_after_axis; in Reshape()
129 size_t dims_[4] = {}; variable
|
| D | embedding_lookup_gpu_kernel.h | 57 …CalEmbeddingLookup(input_addr, indices_addr, output_addr, dims_[0], dims_[1], dims_[2], input_dim1… in Launch()
98 std::fill(dims_, dims_ + 3, 0); in ResetResource()
133 dims_[0] = dim_before_axis; in Reshape()
134 dims_[1] = dim_of_indices; in Reshape()
135 dims_[2] = dim_after_indices; in Reshape()
152 size_t dims_[3] = {}; variable
|
| D | gatherv2_gpu_kernel.h | 62 GatherV2(input_addr, indices_addr, output_addr, dims_[0], dims_[1], dims_[2], input_dim1, in Launch()
105 std::fill(dims_, dims_ + 3, 0); in ResetResource()
143 dims_[0] = dim_before_axis; in Reshape()
144 dims_[1] = dim_of_indices; in Reshape()
145 dims_[2] = dim_after_indices; in Reshape()
152 size_t dims_[3] = {}; variable
|
| D | gathernd_gpu_kernel.h | 69 … GatherNd(input_addr, indices_addr, output_addr, dims_[0], dims_[1], dims_[2], dev_batch_strides_, in Launch()
93 size_t dim_indices_last = dims_[dims_.size() - 1]; in Init()
99 batch_indices_[dim_indices_last - 1] = dims_[1]; in Init()
148 dims_.emplace_back(dim_of_indices); in Reshape()
149 dims_.emplace_back(dim_after_indices); in Reshape()
150 dims_.emplace_back(dim_indices_last); in Reshape()
158 std::vector<size_t> dims_; variable
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/ |
| D | transpose_fp32_coder.cc | 128 dims_ = output_tensor_->shape().size(); in DoCode()
129 if (dims_ > MAX_TRANSPOSE_DIM_SIZE) { in DoCode()
130 int *dim_size = reinterpret_cast<int *>(malloc(dims_ * sizeof(int))); in DoCode()
134 *(dim_size + dims_ - 1) = 1; in DoCode()
135 for (int i = dims_ - 1; i > 0; --i) { in DoCode()
138 code.CodeArray("dim_size", dim_size, dims_); in DoCode()
139 int *position = reinterpret_cast<int *>(malloc(dims_ * thread_num_ * sizeof(int))); in DoCode()
144 code.CodeArray("position", position, dims_ * thread_num_); in DoCode()
|
| D | transpose_fp32_coder.h | 48 int dims_{0};
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | parallel_fusion.h | 41 : nodes_(nodes), dims_(dims), fusion_info_(fusion_info) {} in ParallelInfo()
44 dims_ = obj.dims_; in ParallelInfo()
50 if (nodes_.size() != dims_.size()) { in GetSize()
56 const std::vector<DimInfoPtr> &dims() const { return dims_; } in dims()
61 std::vector<DimInfoPtr> dims_; variable
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/image/lite_cv/ |
| D | lite_mat.cc | 35 dims_ = 0; in LiteMat()
50 dims_ = 0; in LiteMat()
66 dims_ = 0; in LiteMat()
82 dims_ = 0; in LiteMat()
98 dims_ = 0; in LiteMat()
114 dims_ = 0; in LiteMat()
138 dims_ = m.dims_; in LiteMat()
171 dims_ = m.dims_; in operator =()
184 dims_ = 1; in Init()
205 dims_ = 2; in Init()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | unsqueeze_infer.c | 60 if (ax_itr < (size_t)(dim_rank) && param->dims_[ax_itr] == (int)(i)) { in UnsqueezeInferShape()
63 } else if (ax_itr < (size_t)(dim_rank) && param->dims_[ax_itr] + sz == i) { in UnsqueezeInferShape()
|
| D | tile_infer.c | 31 multiples_size_tmp[param->dims_[i]] = param->multiples_[i]; in TileParamCaffe2Tflite()
79 int *dims = param->dims_; in TileInferShape()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/ |
| D | fill_parameter.h | 24 int dims_[COMM_SHAPE_SIZE]; member
|
| D | unsqueeze_parameter.h | 34 int dims_[COMM_SHAPE_SIZE]; member
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/ |
| D | tile_base.h | 26 int dims_[MAX_TILE_DIM_SIZE]; member
|
| /third_party/mindspore/mindspore/lite/src/ops/populate/ |
| D | tile_populate.cc | 47 param->dims_[i] = static_cast<int>(dims->Get(i)); in PopulateTileParameter()
|
| /third_party/mindspore/mindspore/lite/src/ops/populate/v0/ |
| D | unsqueeze_populate_v0.cc | 53 unsqueeze_param->dims_[i++] = *iter; in PopulateUnsqueezeParameter()
|