Searched refs:max_dims (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | addn_infer.c | 37 size_t max_dims = input->shape_size_; in AddnInferShape() local
42 if (inputs[i]->shape_size_ > max_dims) { in AddnInferShape()
43 max_dims = inputs[i]->shape_size_; in AddnInferShape()
51 …if ((inputs[i]->shape_size_ != max_dims) && (GetElementNum(inputs[i]) != GetElementNum(inputs[max_… in AddnInferShape()
59 size_t shift = max_dims - (size_t)(inputs[i]->shape_size_); in AddnInferShape()
|
| /third_party/boost/libs/mpi/src/ |
| D | cartesian_communicator.cpp | 73 int const max_dims = comm.ndims(); in cartesian_communicator() local
75 assert(nbkept <= max_dims); in cartesian_communicator()
76 std::vector<int> bitset(max_dims, int(false)); in cartesian_communicator()
78 assert(keep[i] < max_dims); in cartesian_communicator()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | utils.cc | 58 ShapeVector max_dims(dims.size()); in CalculateDynamicShape() local
63 min_dims[i] = max_dims[i] = dims[i]; in CalculateDynamicShape()
68 max_dims[i] = std::max(shape1->shape()[i], shape2->shape()[i]); in CalculateDynamicShape()
77 max_dims[i] = std::max(shape1->max_shape()[i], shape2->shape()[i]); in CalculateDynamicShape()
86 max_dims[i] = std::max(shape1->shape()[i], shape2->max_shape()[i]); in CalculateDynamicShape()
99 max_dims[i] = std::max(shape1->max_shape()[i], shape2->max_shape()[i]); in CalculateDynamicShape()
101 return std::make_shared<Shape>(dims, min_dims, max_dims); in CalculateDynamicShape()
|
| /third_party/mesa3d/src/gallium/drivers/freedreno/ |
| D | freedreno_state.c | 355 const float max_dims = ctx->screen->gen >= 4 ? 16384.f : 4096.f; in fd_set_viewport_states() local
358 scissor->minx = CLAMP(minx, 0.f, max_dims); in fd_set_viewport_states()
359 scissor->miny = CLAMP(miny, 0.f, max_dims); in fd_set_viewport_states()
360 scissor->maxx = CLAMP(ceilf(maxx), 0.f, max_dims); in fd_set_viewport_states()
361 scissor->maxy = CLAMP(ceilf(maxy), 0.f, max_dims); in fd_set_viewport_states()
|