Home
last modified time | relevance | path

Searched defs:nt_impl (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/
H A DNestedTensorFactories.cpp209 auto *nt_impl = get_nested_tensor_impl(self); in narrow_nested_symint() local
234 auto* nt_impl = get_nested_tensor_impl(self); in alias_nested() local
H A DNestedTensorMath.h16 auto* nt_impl = get_nested_tensor_impl(nt); in map_nt() local
H A DNestedTensorMatmul.cpp192 auto* nt_impl = get_nested_tensor_impl(nt); in matmul_nested_with_broadcasted_dense() local
H A DNestedTensorTransformerFunctions.cpp214 auto* nt_impl = get_nested_tensor_impl(nt); in NestedTensor_to_mask() local
H A DNestedTensorMath.cpp1034 auto *nt_impl = get_nested_tensor_impl(t); in cat_nested_as_jagged() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/cuda/
H A DNestedTensorTransformerUtils.cpp17 auto* nt_impl = get_nested_tensor_impl(nestedtensor); in get_nnz() local