Home
last modified time | relevance | path

Searched refs:dims (Results 1 – 25 of 356) sorted by relevance

12345678910>>...15

/third_party/mindspore/mindspore/lite/mindir/src/
Dmindir_tensor.cc29 std::vector<int32_t> dims(1, 0); in MindIR_Tensor_Create() local
41 …MindIR_Tensor_Create(const std::string &name, DataType data_type, const std::vector<int32_t> &dims, in MindIR_Tensor_Create() argument
47 schema::CreateTensor(fbb, 0, data_type, fbb.CreateVector(dims.data(), dims.size()), in MindIR_Tensor_Create()
74 flatbuffers::Offset<flatbuffers::Vector<int32_t>> dims; in MindIR_Tensor_SetName() local
75 if (value->dims() == nullptr || value->dims()->size() <= 0) { in MindIR_Tensor_SetName()
76 dims = 0; in MindIR_Tensor_SetName()
78 dims = fbb.CreateVector(value->dims()->data(), value->dims()->size()); in MindIR_Tensor_SetName()
87 fbb, 0, value->dataType(), dims, static_cast<schema::Format>(value->format()), 0, 0, data, in MindIR_Tensor_SetName()
116 flatbuffers::Offset<flatbuffers::Vector<int32_t>> dims; in MindIR_Tensor_SetDataType() local
117 if (value->dims() == nullptr || value->dims()->size() <= 0) { in MindIR_Tensor_SetDataType()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/
Dcommon_func.c27 int Offset4d(const int *shape, const int *dims) { return Offset(shape, dims[0], dims[1], dims[2], d… in Offset4d() argument
29 int Offset6d(const int *shape, const int *dims) { in Offset6d() argument
30 return ((OffsetComm(shape, dims[0], dims[1], dims[2]) + dims[3]) * shape[4] + dims[4]) * shape[5]; in Offset6d()
/third_party/skia/tests/
DAndroidCodecTest.cpp71 const auto dims = codec->getInfo().dimensions(); in DEF_TEST() local
73 plus(dims, -1), in DEF_TEST()
74 times(dims, .15f), in DEF_TEST()
75 times(dims, .6f), in DEF_TEST()
76 { (int32_t) (dims.width() * .25f), (int32_t) (dims.height() * .75f ) }, in DEF_TEST()
81 { dims.width(), dims.height() - 1 }, in DEF_TEST()
91 REPORTER_ASSERT(r, size == dims); in DEF_TEST()
94 if (size.width() >= dims.width() || size.height() >= dims.height()) { in DEF_TEST()
97 file, computedSampleSize, dims.width(), dims.height(), in DEF_TEST()
102 REPORTER_ASSERT(r, size.width() < dims.width() && in DEF_TEST()
[all …]
/third_party/ffmpeg/tests/dnn/
Ddnn-layer-mathbinary-test.c67 operands[0].dims[0] = 1; in test_broadcast_input0()
68 operands[0].dims[1] = 1; in test_broadcast_input0()
69 operands[0].dims[2] = 2; in test_broadcast_input0()
70 operands[0].dims[3] = 3; in test_broadcast_input0()
107 operands[0].dims[0] = 1; in test_broadcast_input1()
108 operands[0].dims[1] = 1; in test_broadcast_input1()
109 operands[0].dims[2] = 2; in test_broadcast_input1()
110 operands[0].dims[3] = 3; in test_broadcast_input1()
149 operands[0].dims[0] = 1; in test_no_broadcast()
150 operands[0].dims[1] = 1; in test_no_broadcast()
[all …]
Ddnn-layer-pad-test.c75 operands[0].dims[0] = 1; in test_with_mode_symmetric()
76 operands[0].dims[1] = 4; in test_with_mode_symmetric()
77 operands[0].dims[2] = 4; in test_with_mode_symmetric()
78 operands[0].dims[3] = 3; in test_with_mode_symmetric()
140 operands[0].dims[0] = 3; in test_with_mode_reflect()
141 operands[0].dims[1] = 2; in test_with_mode_reflect()
142 operands[0].dims[2] = 2; in test_with_mode_reflect()
143 operands[0].dims[3] = 3; in test_with_mode_reflect()
206 operands[0].dims[0] = 1; in test_with_mode_constant()
207 operands[0].dims[1] = 2; in test_with_mode_constant()
[all …]
Ddnn-layer-avgpool-test.c87 operands[0].dims[0] = 1; in test_with_same()
88 operands[0].dims[1] = 5; in test_with_same()
89 operands[0].dims[2] = 6; in test_with_same()
90 operands[0].dims[3] = 3; in test_with_same()
167 operands[0].dims[0] = 1; in test_with_valid()
168 operands[0].dims[1] = 5; in test_with_valid()
169 operands[0].dims[2] = 6; in test_with_valid()
170 operands[0].dims[3] = 3; in test_with_valid()
Ddnn-layer-conv2d-test.c114 operands[0].dims[0] = 1; in test_with_same_dilate()
115 operands[0].dims[1] = 5; in test_with_same_dilate()
116 operands[0].dims[2] = 6; in test_with_same_dilate()
117 operands[0].dims[3] = 3; in test_with_same_dilate()
218 operands[0].dims[0] = 1; in test_with_valid()
219 operands[0].dims[1] = 5; in test_with_valid()
220 operands[0].dims[2] = 6; in test_with_valid()
221 operands[0].dims[3] = 3; in test_with_valid()
/third_party/vk-gl-cts/external/vulkancts/modules/vulkan/compute/
DvktComputeCooperativeMatrixTests.cpp311 } dims[4]; in initPrograms() local
316 dims[0].rows = "M"; in initPrograms()
317 dims[0].cols = "K"; in initPrograms()
318 dims[1].rows = "K"; in initPrograms()
319 dims[1].cols = "N"; in initPrograms()
320 dims[2].rows = "M"; in initPrograms()
321 dims[2].cols = "N"; in initPrograms()
322 dims[3].rows = "M"; in initPrograms()
323 dims[3].cols = "N"; in initPrograms()
327 dims[0].rows = "M"; in initPrograms()
[all …]
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/
Dbiasadd_fp32_coder.cc52 std::vector<int> dims = input_tensor_->shape(); in DoCode() local
54 arithmetic_parameter_->ndim_ = dims.size(); in DoCode()
56 MS_CHECK_TRUE(dims.size() <= DIMENSION_10D, "dims.size() must not be greater than 10!"); in DoCode()
57 for (size_t i = 0; i < dims.size(); i++) { in DoCode()
58 arithmetic_parameter_->in_shape0_[i] = dims[i]; in DoCode()
62 for (size_t i = 0; i < dims.size(); i++) { in DoCode()
63 if (i == dims.size() - 1) { in DoCode()
64 arithmetic_parameter_->in_shape1_[i] = dims[dims.size() - 1]; in DoCode()
71 for (size_t i = 0; i < dims.size(); i++) { in DoCode()
72 arithmetic_parameter_->out_shape_[i] = dims[i]; in DoCode()
/third_party/boost/libs/mpi/src/
Dcartesian_communicator.cpp48 std::vector<int> dims(topology.size()); in cartesian_communicator() local
52 dims[i] = topology[i].size; in cartesian_communicator()
56 if (std::count(dims.begin(), dims.end(), 0) > 0) { in cartesian_communicator()
57 cartesian_dimensions(comm, dims); in cartesian_communicator()
61 ((MPI_Comm)comm, dims.size(), in cartesian_communicator()
62 detail::c_data(dims), detail::c_data(periodic), in cartesian_communicator()
148 cartesian_topology::split(std::vector<int>& dims, std::vector<bool>& periodics) const { in split() argument
150 dims.resize(ndims); in split()
154 dims[i] = d.size; in split()
160 cartesian_dimensions(int sz, std::vector<int>& dims) { in cartesian_dimensions() argument
[all …]
/third_party/mindspore/mindspore/ccsrc/cxx_api/model/ms/
Dms_model.cc32 static std::string GenerateShapeKey(const std::vector<std::vector<int64_t>> &dims) { in GenerateShapeKey() argument
34 for (size_t i = 0; i < dims.size(); ++i) { in GenerateShapeKey()
36 for (size_t j = 0; j < dims[i].size(); ++j) { in GenerateShapeKey()
37 shape_key += std::to_string(dims[i][j]); in GenerateShapeKey()
38 if (j + 1 < dims[i].size()) { in GenerateShapeKey()
42 if (i + 1 < dims.size()) { in GenerateShapeKey()
49 std::shared_ptr<GraphCell> MsModel::GenerateGraphCell(const std::vector<std::vector<int64_t>> &dims in GenerateGraphCell() argument
50 std::string shape_key = GenerateShapeKey(dims); in GenerateGraphCell()
60 if (dims.size() != inputs.size()) { in GenerateGraphCell()
61 …MS_LOG(ERROR) << "Invalid dims size " << dims.size() << " not match model inputs size " << inputs.… in GenerateGraphCell()
[all …]
/third_party/mindspore/mindspore/lite/src/delegate/tensorrt/
Dtensorrt_utils.cc23 nvinfer1::Dims dims{}; in ConvertCudaDims() local
24 if (size > static_cast<size_t>(dims.MAX_DIMS)) { in ConvertCudaDims()
26 return dims; in ConvertCudaDims()
28 dims.nbDims = size; in ConvertCudaDims()
30 dims.d[i] = data; in ConvertCudaDims()
32 return dims; in ConvertCudaDims()
36 nvinfer1::Dims dims{}; in ConvertCudaDims() local
37 if (size > static_cast<int64_t>(dims.MAX_DIMS)) { in ConvertCudaDims()
39 return dims; in ConvertCudaDims()
41 dims.nbDims = size; in ConvertCudaDims()
[all …]
/third_party/flutter/skia/tests/
DAndroidCodecTest.cpp71 const auto dims = codec->getInfo().dimensions(); in DEF_TEST() local
73 plus(dims, -1), in DEF_TEST()
74 times(dims, .15f), in DEF_TEST()
75 times(dims, .6f), in DEF_TEST()
76 { (int32_t) (dims.width() * .25f), (int32_t) (dims.height() * .75f ) }, in DEF_TEST()
81 { dims.width(), dims.height() - 1 }, in DEF_TEST()
91 REPORTER_ASSERT(r, size == dims); in DEF_TEST()
94 if (size.width() >= dims.width() || size.height() >= dims.height()) { in DEF_TEST()
97 file, computedSampleSize, dims.width(), dims.height(), in DEF_TEST()
102 REPORTER_ASSERT(r, size.width() < dims.width() && in DEF_TEST()
[all …]
/third_party/mesa3d/src/mesa/main/
Dtexstorage.c52 legal_texobj_target(const struct gl_context *ctx, GLuint dims, GLenum target) in legal_texobj_target() argument
54 if (dims < 1 || dims > 3) { in legal_texobj_target()
55 _mesa_problem(ctx, "invalid dims=%u in legal_texobj_target()", dims); in legal_texobj_target()
59 switch (dims) { in legal_texobj_target()
83 switch (dims) { in legal_texobj_target()
308 GLuint dims, GLenum target, in tex_storage_error_check() argument
323 suffix, dims); in tex_storage_error_check()
331 "glTex%sStorage%dD(internalformat = %s)", suffix, dims, in tex_storage_error_check()
340 suffix, dims); in tex_storage_error_check()
348 suffix, dims); in tex_storage_error_check()
[all …]
Dtexstore.c153 (void) dims; in _mesa_texstore_z32()
169 const GLvoid *src = _mesa_image_address(dims, srcPacking, in _mesa_texstore_z32()
190 (void) dims; in _mesa_texstore_x8_z24()
199 const GLvoid *src = _mesa_image_address(dims, srcPacking, in _mesa_texstore_x8_z24()
220 (void) dims; in _mesa_texstore_z24_x8()
229 const GLvoid *src = _mesa_image_address(dims, srcPacking, in _mesa_texstore_z24_x8()
253 (void) dims; in _mesa_texstore_z16()
263 const GLvoid *src = _mesa_image_address(dims, srcPacking, in _mesa_texstore_z16()
283 (void) ctx; (void) dims; (void) baseInternalFormat; in _mesa_texstore_ycbcr()
295 _mesa_memcpy_texture(ctx, dims, in _mesa_texstore_ycbcr()
[all …]
/third_party/mindspore/mindspore/lite/src/ops/populate/
Dtile_populate.cc39 auto dims = value->dims(); in PopulateTileParameter() local
40 if (dims != nullptr) { in PopulateTileParameter()
41 if (dims->size() > MAX_SHAPE_SIZE) { in PopulateTileParameter()
42 MS_LOG(ERROR) << "Invalid dims size " << dims->size(); in PopulateTileParameter()
46 for (size_t i = 0; i < dims->size(); ++i) { in PopulateTileParameter()
47 param->dims_[i] = static_cast<int>(dims->Get(i)); in PopulateTileParameter()
49 param->dims_size_ = dims->size(); in PopulateTileParameter()
/third_party/boost/boost/mpi/
Dcartesian_communicator.hpp138 cartesian_topology(std::vector<cartesian_dimension> const& dims) in cartesian_topology() argument
139 : super(dims) {} in cartesian_topology()
146 explicit cartesian_topology(InitArr dims) in cartesian_topology() argument
148 BOOST_FOREACH(cartesian_dimension const& d, dims) { in cartesian_topology()
157 explicit cartesian_topology(std::initializer_list<cartesian_dimension> dims) in cartesian_topology() argument
158 : super(dims) {} in cartesian_topology()
165 explicit cartesian_topology(cartesian_dimension (&dims)[NDIM]) in cartesian_topology()
166 : super(dims, dims+NDIM) {} in cartesian_topology()
220 void split(std::vector<int>& dims, std::vector<bool>& periodics) const;
323 const cartesian_topology& dims,
[all …]
/third_party/mindspore/mindspore/lite/src/ops/populate/v0/
Dtile_populate_v0.cc39 if (tile_prim->dims() != nullptr) { in PopulateTileParameter()
40 auto dims = tile_prim->dims(); in PopulateTileParameter() local
41 if (dims == nullptr) { in PopulateTileParameter()
46 if (dims->size() > MAX_TILE_DIM_SIZE) { in PopulateTileParameter()
51 for (size_t i = 0; i < dims->size(); i++) { in PopulateTileParameter()
52 tile_param->dims_[i] = static_cast<int>(dims->Get(i)); in PopulateTileParameter()
54 tile_param->dims_size_ = dims->size(); in PopulateTileParameter()
/third_party/boost/boost/gil/
Dimage.hpp239 void recreate(const point_t& dims, std::size_t alignment = 0) in recreate() argument
241 if (dims == _view.dimensions() && _align_in_bytes == alignment) in recreate()
246 if (_allocated_bytes >= total_allocated_size_in_bytes(dims)) in recreate()
249 create_view(dims, std::integral_constant<bool, IsPlanar>()); in recreate()
254 image tmp(dims, alignment); in recreate()
264 void recreate(const point_t& dims, const Pixel& p_in, std::size_t alignment = 0) in recreate() argument
266 if (dims == _view.dimensions() && _align_in_bytes == alignment) in recreate()
271 if (_allocated_bytes >= total_allocated_size_in_bytes(dims)) in recreate()
274 create_view(dims, typename std::integral_constant<bool, IsPlanar>()); in recreate()
279 image tmp(dims, p_in, alignment); in recreate()
[all …]
/third_party/skia/third_party/externals/tint/src/reader/spirv/
Dparser_type.cc84 return utils::Hash(t.dims, t.type); in operator ()()
90 return utils::Hash(t.dims, t.type); in operator ()()
96 return utils::Hash(t.dims, t.format, t.access); in operator ()()
123 return a.dims == b.dims && a.type == b.type; in operator ==()
127 return a.dims == b.dims && a.type == b.type; in operator ==()
131 return a.dims == b.dims && a.format == b.format; in operator ==()
204 Texture::Texture(ast::TextureDimension d) : dims(d) {} in Texture()
211 return b.ty.depth_texture(dims); in Build()
220 return b.ty.depth_multisampled_texture(dims); in Build()
228 return b.ty.multisampled_texture(dims, type->Build(b)); in Build()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/
Dsplit_gpu_kernel.h65 int dims = SizeToInt(input_shape.size()); in Init() local
67 if (axis_ < -dims || axis_ >= dims) { in Init()
71 axis_ += dims; in Init()
141 int dims = SizeToInt(input_shape.size()); in CheckParam() local
151 if (dims == 0) { in CheckParam()
152 MS_LOG(ERROR) << "Input dims is " << dims << ", scalar is not supported."; in CheckParam()
155 if (axis_ < -dims || axis_ >= dims) { in CheckParam()
156 MS_LOG(ERROR) << "Attr axis " << axis_ << " must be in " << -dims << "~" << dims; in CheckParam()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/
Dtile_infer.c79 int *dims = param->dims_; in TileInferShape() local
87 ShapePush(dims, &dims_size, dim); in TileInferShape()
98 if (dims[i] >= MAX_SHAPE_SIZE || input->shape_[dims[i]] == 0) { 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/ffmpeg/libavfilter/dnn/
Ddnn_backend_native_layer_depth2space.c59 int number = operands[input_operand_index].dims[0]; in ff_dnn_execute_layer_depth2space()
60 int height = operands[input_operand_index].dims[1]; in ff_dnn_execute_layer_depth2space()
61 int width = operands[input_operand_index].dims[2]; in ff_dnn_execute_layer_depth2space()
62 int channels = operands[input_operand_index].dims[3]; in ff_dnn_execute_layer_depth2space()
72 output_operand->dims[0] = number; in ff_dnn_execute_layer_depth2space()
73 output_operand->dims[1] = height * block_size; in ff_dnn_execute_layer_depth2space()
74 output_operand->dims[2] = width * block_size; in ff_dnn_execute_layer_depth2space()
75 output_operand->dims[3] = new_channels; in ff_dnn_execute_layer_depth2space()
Ddnn_backend_openvino.c126 dimensions_t dims; in fill_model_input_ov() local
141 status |= ie_blob_get_dims(input_blob, &dims); in fill_model_input_ov()
156 input.height = dims.dims[2]; in fill_model_input_ov()
157 input.width = dims.dims[3]; in fill_model_input_ov()
158 input.channels = dims.dims[1]; in fill_model_input_ov()
165 av_assert0(request->task_count <= dims.dims[0]); in fill_model_input_ov()
185 dimensions_t dims; in infer_completion_callback() local
221 status |= ie_blob_get_dims(output_blob, &dims); in infer_completion_callback()
229 output.channels = dims.dims[1]; in infer_completion_callback()
230 output.height = dims.dims[2]; in infer_completion_callback()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/
Dutils.h25 static inline size_t GetInputOffset(int num_dims, const int *dims, const int *iter) { in GetInputOffset() argument
28 offset = offset * (size_t)(dims[idx]) + (size_t)(iter[idx]); in GetInputOffset()
34 static inline size_t GetOutputOffset(int num_dims, const int *dims, const int *iter, int num_axis, … in GetOutputOffset() argument
47 offset = offset * (size_t)(dims[idx]) + (size_t)(iter[idx]); in GetOutputOffset()
53 static inline int NextIndex(int num_dims, const int *dims, int *current) { in NextIndex() argument
57 if (dims[idx] == current_val) { in NextIndex()

12345678910>>...15