Home
last modified time | relevance | path

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

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