Searched defs:max_dims (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | SpectralOps.cpp | 376 const auto max_dims = std::min(static_cast<size_t>(cufft_max_ndim), sorted_dims.size()); in _fft_r2c_cufft() local
476 const auto max_dims = std::min(static_cast<size_t>(cufft_max_ndim), sorted_dims.size()); in _fft_c2c_cufft() local
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/ |
| H A D | Validation.h | 173 const auto max_dims = std::max(dimensions1.num_dimensions(), dimensions2.num_dimensions()); variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/versioning/ |
| H A D | op_version.cc | 43 int max_dims = 0; in GetInputMaxDims() local
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | hypothesis_utils.py | 162 def array_shapes(draw, min_dims=1, max_dims=None, min_side=1, max_side=None, max_numel=None): argument
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/freedreno/ |
| H A D | freedreno_state.c | 421 const float max_dims = ctx->screen->gen >= 4 ? 16384.f : 4096.f; in fd_set_viewport_states() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | shape_analysis.cpp | 777 auto max_dims = any_type->dim(); in PropagateTensorShapeOnNode() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_nestedtensor.py | 146 max_dims, argument
852 def random_nt_pair(self, device, dtype, num_tensors, max_dims): argument
|
| H A D | test_linalg.py | 3904 min_dims=1, max_dims=3, # min and max number of dimensions per operand argument
|