Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/
DLegacyBatchedFallback.cpp211 static Tensor safeStack(TensorList tensors) { in safeStack() function
361 auto flat_output = safeStack(shards); in batchedTensorForLoopFallback()
/external/pytorch/aten/src/ATen/functorch/
DBatchedFallback.cpp217 static Tensor safeStack(TensorList tensors) { in safeStack() function
382 auto flat_output = safeStack(shards); in batchedTensorForLoopFallback()