Home
last modified time | relevance | path

Searched refs:batchedTensorForLoopFallback (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchedFallback.h32 void batchedTensorForLoopFallback(const c10::OperatorHandle& op, torch::jit::Stack* stack);
62 batchedTensorForLoopFallback(op, &stack); in slow_fallback()
69 batchedTensorForLoopFallback(op, &stack); in slow_fallback()
76 batchedTensorForLoopFallback(op, &stack); in slow_fallback()
H A DBatchedFallback.cpp259 void batchedTensorForLoopFallback(const c10::OperatorHandle& op, torch::jit::Stack* stack) { in batchedTensorForLoopFallback() function
H A DLegacyBatchingRegistrations.cpp732 m.fallback(torch::CppFunction::makeFromBoxedFunction<&batchedTensorForLoopFallback>()); in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DLegacyBatchedFallback.h21 void batchedTensorForLoopFallback(
H A DLegacyBatchedFallback.cpp250 void batchedTensorForLoopFallback(const c10::OperatorHandle& op, torch::jit::Stack* stack) { in batchedTensorForLoopFallback() function
H A DLegacyBatchingRegistrations.cpp1076 m.fallback(torch::CppFunction::makeFromBoxedFunction<&batchedTensorForLoopFallback>()); in TORCH_LIBRARY_IMPL()