Home
last modified time | relevance | path

Searched refs:kVmapTransformStaticInputSize (Results 1 – 4 of 4) sorted by relevance

/external/pytorch/aten/src/ATen/
DLegacyVmapTransforms.h34 constexpr int64_t kVmapTransformStaticInputSize = 4; variable
36 SmallVector<VmapPhysicalView, kVmapTransformStaticInputSize>;
DLegacyBatchedFallback.cpp111 SmallVector<Tensor,kVmapTransformStaticInputSize> batched_tensor_inputs; in batchedTensorInplaceForLoopFallback()
275 SmallVector<Tensor,kVmapTransformStaticInputSize> batched_tensor_inputs; in batchedTensorForLoopFallback()
/external/pytorch/aten/src/ATen/functorch/
DLegacyVmapTransforms.h42 constexpr int64_t kVmapTransformStaticInputSize = 4; variable
43 using VmapPhysicalViewVec = SmallVector<VmapPhysicalView, kVmapTransformStaticInputSize>;
DBatchedFallback.cpp116 at::SmallVector<Tensor,kVmapTransformStaticInputSize> batched_tensor_inputs; in batchedTensorInplaceForLoopFallback()
291 at::SmallVector<Tensor,kVmapTransformStaticInputSize> batched_tensor_inputs; in batchedTensorForLoopFallback()
432 at::SmallVector<Tensor,kVmapTransformStaticInputSize> batched_tensor_inputs; in batchedNestedTensorForLoopFallback()