Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchedTensorImpl.h125 inline BatchedTensorImpl* maybeGetBatchedImpl(const Tensor& tensor) { in maybeGetBatchedImpl() function
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DLegacyBatchedTensorImpl.h119 inline BatchedTensorImpl* maybeGetBatchedImpl(const Tensor& tensor) { in maybeGetBatchedImpl() function
/aosp_15_r20/external/pytorch/functorch/csrc/dim/
H A Ddim.cpp762 at::functorch::BatchedTensorImpl* maybeGetBatchedImpl(const at::Tensor& tensor) { in maybeGetBatchedImpl() function