Searched refs:kVmapTransformStaticInputSize (Results 1 – 4 of 4) sorted by relevance
34 constexpr int64_t kVmapTransformStaticInputSize = 4; variable36 SmallVector<VmapPhysicalView, kVmapTransformStaticInputSize>;
111 SmallVector<Tensor,kVmapTransformStaticInputSize> batched_tensor_inputs; in batchedTensorInplaceForLoopFallback()275 SmallVector<Tensor,kVmapTransformStaticInputSize> batched_tensor_inputs; in batchedTensorForLoopFallback()
42 constexpr int64_t kVmapTransformStaticInputSize = 4; variable43 using VmapPhysicalViewVec = SmallVector<VmapPhysicalView, kVmapTransformStaticInputSize>;
116 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()