Home
last modified time | relevance | path

Searched defs:size_tensor_stride (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/cuda/
H A DNestedTensorTransformerUtils.cpp19 auto size_tensor_stride = sizes.stride(0); in get_nnz() local
46 auto size_tensor_stride = sizes.stride(0); in cumulative_and_max_seq_len_nnz() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/
H A Dsdp_utils_cpp.h168 const int64_t size_tensor_stride = sizes.stride(0); in check_for_seq_len_0_and_consistent_head_dim_nested_tensor_helper() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/
H A Dsdp_utils.cpp744 const int64_t size_tensor_stride = sizes.stride(0); in check_for_seq_len_1_nested_tensor() local