Home
last modified time | relevance | path

Searched defs:input_batched (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DLegacyBatchingRegistrations.cpp498 auto* input_batched = unsafeGetBatchedImpl(input); in unwrap_and_call() local
509 auto* input_batched = unsafeGetBatchedImpl(input); in unwrap_and_call_method() local
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DLegacyBatchingRegistrations.cpp745 auto* input_batched = unsafeGetBatchedImpl(input); in unwrap_and_call() local
753 auto* input_batched = unsafeGetBatchedImpl(input); in unwrap_and_call_method() local
972 auto* input_batched = unsafeGetBatchedImpl(self); in to_dtype_layout_batching_rule() local