Home
last modified time | relevance | path

Searched refs:batchedTensorInplaceForLoopFallback (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DLegacyBatchedFallback.cpp93 static void batchedTensorInplaceForLoopFallback(const c10::OperatorHandle& op, torch::jit::Stack* s… in batchedTensorInplaceForLoopFallback() function
255 batchedTensorInplaceForLoopFallback(op, stack); in batchedTensorForLoopFallback()
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchedFallback.cpp98 static void batchedTensorInplaceForLoopFallback(const c10::OperatorHandle& op, torch::jit::Stack* s… in batchedTensorInplaceForLoopFallback() function
276 batchedTensorInplaceForLoopFallback(op, stack); in batchedTensorForLoopFallback()