Home
last modified time | relevance | path

Searched defs:batched_tensor_inputs (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchedFallback.cpp116 at::SmallVector<Tensor,kVmapTransformStaticInputSize> batched_tensor_inputs; in batchedTensorInplaceForLoopFallback() local
291 at::SmallVector<Tensor,kVmapTransformStaticInputSize> batched_tensor_inputs; in batchedTensorForLoopFallback() local
432 at::SmallVector<Tensor,kVmapTransformStaticInputSize> batched_tensor_inputs; in batchedNestedTensorForLoopFallback() local
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DLegacyBatchedFallback.cpp111 SmallVector<Tensor,kVmapTransformStaticInputSize> batched_tensor_inputs; in batchedTensorInplaceForLoopFallback() local
275 SmallVector<Tensor,kVmapTransformStaticInputSize> batched_tensor_inputs; in batchedTensorForLoopFallback() local