Home
last modified time | relevance | path

Searched refs:kStackArrayMaxDims (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/cuda/
H A DNestedTensorTransformerFunctions.cu528 constexpr size_t kStackArrayMaxDims = 5; variable
532 T vals[kStackArrayMaxDims];
584 TORCH_CHECK(num_jagged_dim <= kStackArrayMaxDims); in check_shape_and_partition_()