Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/ComputeLibrary/src/core/
H A DValidate.cpp71 … const arm_compute::Coordinates &pos, unsigned int max_dim) in error_on_coordinates_dimensions_gte()
81 … const arm_compute::Window &win, unsigned int max_dim) in error_on_window_dimensions_gte()
/aosp_15_r20/external/executorch/backends/cadence/hifi/operators/
H A Dop_div.cpp81 int max_dim = a.dim() > b.dim() ? a.dim() : b.dim(); in div_out() local
195 int max_dim = a.dim() > b.dim() ? a.dim() : b.dim(); in div_out_mode() local
H A Dop_mul.cpp102 int max_dim = a.dim() > b.dim() ? a.dim() : b.dim(); in mul_out() local
H A Dop_where.cpp64 int max_dim = a.dim() > b.dim() ? a.dim() : b.dim(); in where_out() local
H A Dop_add.cpp133 int max_dim = a.dim() > b.dim() ? a.dim() : b.dim(); in add_out() local
H A Dop_sub.cpp128 int max_dim = a.dim() > b.dim() ? a.dim() : b.dim(); in sub_out() local
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DCPUApplyUtils.h215 inline int64_t max_dim() { in max_dim() function
220 inline int64_t max_dim(Arg& iter, Args&... iter_tail) { in max_dim() function
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/operators/
H A Dreduction.cpp150 auto max_dim = std::get<int64_t>(inputs[1]); in computeMax() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Deigen_mkldnn_contraction_kernel_test.cc31 Eigen::array<Index, NumDims> RandomDims(int min_dim = 1, int max_dim = 20) { in RandomDims()
/aosp_15_r20/external/mesa3d/src/nouveau/vulkan/
H A Dnvk_upload_queue.c355 const uint32_t max_dim = 1 << 17; in nvk_upload_queue_fill_locked() local
H A Dnvk_cmd_copy.c728 uint32_t max_dim = 1 << 15; in nvk_CmdFillBuffer() local
H A Dnvk_image.c403 const uint32_t max_dim = in nvk_GetPhysicalDeviceImageFormatProperties2() local
/aosp_15_r20/external/executorch/kernels/portable/cpu/util/
H A Dreduce_util.cpp66 const size_t max_dim, in check_dim_in_dim_list()
/aosp_15_r20/external/pytorch/test/
H A Dtest_shape_ops.py584 def _test_fliplr_flipud(self, torch_fn, np_fn, min_dim, max_dim, device, dtype): argument
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DConvUtils.h147 constexpr int max_dim = 3; variable
/aosp_15_r20/external/OpenCL-CTS/test_conformance/thread_dimensions/
H A Dtest_thread_dimensions.cpp416 …ontext, cl_command_queue queue, cl_uint dimensions, cl_uint min_dim, cl_uint max_dim, cl_uint quic… in test_thread_dimensions()
/aosp_15_r20/external/mesa3d/src/gallium/drivers/svga/
H A Dsvga_resource_texture.c833 unsigned max_dim = template->width0; in svga_texture_create() local
/aosp_15_r20/external/eigen/unsupported/test/
H A Dcxx11_tensor_executor.cpp49 static array<Index, NumDims> RandomDims(int min_dim = 1, int max_dim = 20) { in RandomDims()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dtensor_shape.cc733 const int max_dim = unknown_rank() ? -1 : dims(); in end() local
/aosp_15_r20/external/mesa3d/src/gallium/drivers/radeonsi/
H A Dsi_test_image_copy_region.c406 unsigned max_dim = MAX3(templ->width0, templ->height0, templ->depth0); in set_random_image_attrs() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dexternal_functions.cpp1315 int64_t max_dim = extra_args[0]; in nnc_aten_max_red() local
1340 int64_t max_dim = extra_args[0]; in nnc_aten_max_red_out() local
/aosp_15_r20/art/runtime/jit/
H A Dprofile_saver.cc683 size_t max_dim = max_primitive_array_dimensions_[i]; in UpdateProfile() local
/aosp_15_r20/external/mesa3d/src/asahi/vulkan/
H A Dhk_image.c378 const uint32_t max_dim = 16384; in hk_GetPhysicalDeviceImageFormatProperties2() local
/aosp_15_r20/external/pytorch/torch/jit/
H A D_shape_functions.py190 def max_dim(self: List[int], dim: int, keep_dim: bool): function
/aosp_15_r20/external/OpenCL-CTS/test_conformance/basic/
H A Dtest_progvar.cpp581 cl_uint max_dim = 0; in l_load_abilities() local

12