Home
last modified time | relevance | path

Searched defs:nt_self (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/
H A DNestedTensorBackward.cpp123 auto nt_self = get_nested_tensor_impl(nested_self); in _nested_sum_backward_cpu() local
168 auto nt_self = get_nested_tensor_impl(nested_self); in _nested_select_backward_symint() local
H A DNestedTensorTransformerFunctions.cpp77 auto* nt_self = get_nested_tensor_impl_or_null(self); in NestedTensor_matmul() local
128 const auto& nt_self = *get_nested_tensor_impl(self); in NestedTensor_add_NestedTensor_in_place() local
H A DNestedTensorFactories.cpp121 const auto* nt_self = get_nested_tensor_impl(self); in copy_nested_() local
H A DNestedTensorBinaryOps.cpp258 const auto& nt_self = *self_impl; in NestedTensor_elementwise__Tensor() local
H A DNestedTensorMath.cpp859 auto* nt_self = get_nested_tensor_impl(self); in values_nested() local