Home
last modified time | relevance | path

Searched defs:max_dim (Results 1 – 24 of 24) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Dceil.cc29 const int64_t max_dim = 8; in InferShape() local
60 const int64_t max_dim = 8; in InferShape() local
Dsinh.cc27 const size_t max_dim = 8; in InferShape() local
Dfloor_div.cc29 const int64_t max_dim = 8; in InferShape() local
Dcosh.cc32 const size_t max_dim = 8; in InferShape() local
Dfloor_mod.cc29 const int64_t max_dim = 8; in InferShape() local
Dsort_ext.cc58 auto max_dim = 0; in InferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/int8/
Ddiv_int8_coder.cc67 size_t max_dim = sizeof(tile_para.in_shape0_) / sizeof(int); in DoCode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/
Daddn_infer.c71 size_t max_dim = 0; in AddnInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/pi/graph_guard/
Dstrategy.cc149 static OptStrategy::CalcKind TensorComputable(PyObject *obj, ssize_t max_dim) { in TensorComputable()
160 static OptStrategy::CalcKind StubTensorComputable(PyObject *obj, ssize_t max_dim) { in StubTensorComputable()
189 static OptStrategy::CalcKind ObjectComputable(PyObject *obj, ssize_t max_dim = kMaxCalcDim) { in ObjectComputable()
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dleft_shift.cc50 const int64_t max_dim = 8; in LeftShiftInferShape() local
Dragged_range.cc89 const int64_t max_dim = 2; in RaggedRangeInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/
Drandom_gamma_grad.cc59 const int64_t max_dim = 8; in RandomGammaGradInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_class/
Dmultilabel_margin_loss_helper.h138 int64_t max_dim = 2; in CheckKernelParam() local
Dmultilabel_margin_loss_grad_helper.h122 int64_t max_dim = 2; in CheckKernelParam() local
/third_party/mesa3d/src/nouveau/vulkan/
Dnvk_upload_queue.c355 const uint32_t max_dim = 1 << 17; in nvk_upload_queue_fill_locked() local
Dnvk_cmd_copy.c745 uint32_t max_dim = 1 << 15; in nvk_CmdFillBuffer() local
Dnvk_image.c415 const uint32_t max_dim = in nvk_GetPhysicalDeviceImageFormatProperties2() local
Dnvk_cmd_draw.c1841 const float max_dim = (float)0xffff; in nvk_flush_vp_state() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/
Dpad.c126 int PadExtendDims(int *dims, const int *origin_dims, int max_dim, int origin_dim, int init_value) { in PadExtendDims()
/third_party/mesa3d/src/gallium/drivers/svga/
Dsvga_resource_texture.c834 unsigned max_dim = template->width0; in svga_texture_create() local
/third_party/mesa3d/src/gallium/drivers/radeonsi/
Dsi_test_image_copy_region.c406 unsigned max_dim = MAX3(templ->width0, templ->height0, templ->depth0); in set_random_image_attrs() local
/third_party/mesa3d/src/asahi/vulkan/
Dhk_image.c396 const uint32_t max_dim = 16384; in hk_GetPhysicalDeviceImageFormatProperties2() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/model/
Dop_node.cc409 auto max_dim = max_dim_input->size(); in BroadcastShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pybind_api/ir/
Dtensor_index_py.cc461 constexpr int64_t max_dim = 1024 * 32; in CheckLargeTensor() local