Home
last modified time | relevance | path

Searched refs:batchedNestedTensorForLoopFallback (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchedFallback.h33 void batchedNestedTensorForLoopFallback(const c10::OperatorHandle& op, torch::jit::Stack* stack);
H A DBatchedFallback.cpp399 void batchedNestedTensorForLoopFallback(const c10::OperatorHandle& op, torch::jit::Stack* stack) { in batchedNestedTensorForLoopFallback() function
H A DLegacyBatchingRegistrations.cpp759 m.fallback(torch::CppFunction::makeFromBoxedFunction<&batchedNestedTensorForLoopFallback>()); in TORCH_LIBRARY_IMPL()