Searched refs:logicalToPhysical (Results 1 – 9 of 9) sorted by relevance
78 auto self_physical = MultiBatchVmapTransform::logicalToPhysical(self); in sum_batching_rule()99 auto physical_args = BroadcastingVmapTransform::logicalToPhysical({self, other}); in binary_pointwise_batching_rule()104 auto other_physical = MultiBatchVmapTransform::logicalToPhysical(other); in binary_pointwise_batching_rule()109 auto self_physical = MultiBatchVmapTransform::logicalToPhysical(self); in binary_pointwise_batching_rule()149 auto physical_args = BroadcastingVmapTransform::logicalToPhysical( in binary_pointwise_batching_rule()156 auto self_physical = MultiBatchVmapTransform::logicalToPhysical(self); in expand_batching_rule()194 auto self_physical = MultiBatchVmapTransform::logicalToPhysical(self); in chunk_batching_rule()202 auto self_physical = MultiBatchVmapTransform::logicalToPhysical(self); in clamp_batching_rule()208 auto self_physical = MultiBatchVmapTransform::logicalToPhysical(self); in clamp_min_batching_rule()214 auto self_physical = MultiBatchVmapTransform::logicalToPhysical(self); in clamp_max_batching_rule()[all …]
59 static VmapPhysicalView logicalToPhysical(const Tensor& logical_tensor);60 static VmapPhysicalViewVec logicalToPhysical(ITensorListRef logical_tensors);83 static VmapPhysicalViewVec logicalToPhysical(TensorList logical_tensors);
43 VmapPhysicalView MultiBatchVmapTransform::logicalToPhysical(const Tensor& logical_tensor) { in logicalToPhysical() function in at::MultiBatchVmapTransform190 MultiBatchVmapTransform::logicalToPhysical(ITensorListRef logical_tensors) { in logicalToPhysical() function in at::MultiBatchVmapTransform256 VmapPhysicalViewVec BroadcastingVmapTransform::logicalToPhysical(TensorList logical_tensors) { in logicalToPhysical() function in at::BroadcastingVmapTransform
163 const auto input_physical_views = MultiBatchVmapTransform::logicalToPhysical( in batchedTensorInplaceForLoopFallback()297 const auto input_physical_views = MultiBatchVmapTransform::logicalToPhysical( in batchedTensorForLoopFallback()
66 static VmapPhysicalView logicalToPhysical(const Tensor& logical_tensor);67 static VmapPhysicalViewVec logicalToPhysical(ITensorListRef logical_tensors);90 static VmapPhysicalViewVec logicalToPhysical(TensorList logical_tensors);
242 auto self_physical = MultiBatchVmapTransform::logicalToPhysical(self); in split_batching_rule()254 auto self_physical = MultiBatchVmapTransform::logicalToPhysical(self); in split_with_sizes_batching_rule()266 auto self_physical = MultiBatchVmapTransform::logicalToPhysical(self); in split_with_sizes_copy_batching_rule()278 auto self_physical = MultiBatchVmapTransform::logicalToPhysical(self); in unbind_batching_rule()369 auto physical_view = MultiBatchVmapTransform::logicalToPhysical(tensor); in as_strided_batching_rule()586 auto physical_views = MultiBatchVmapTransform::logicalToPhysical(tensors); in block_diag_batching_rule()614 auto physical_views = MultiBatchVmapTransform::logicalToPhysical(tensors); in stack_batching_rule()644 auto physical_view = MultiBatchVmapTransform::logicalToPhysical(self); in new_empty_strided_batching_rule()
37 VmapPhysicalView MultiBatchVmapTransform::logicalToPhysical(const Tensor& logical_tensor) { in logicalToPhysical() function in at::functorch::MultiBatchVmapTransform120 MultiBatchVmapTransform::logicalToPhysical(ITensorListRef logical_tensors) { in logicalToPhysical() function in at::functorch::MultiBatchVmapTransform172 VmapPhysicalViewVec BroadcastingVmapTransform::logicalToPhysical(TensorList logical_tensors) { in logicalToPhysical() function in at::functorch::BroadcastingVmapTransform
168 const auto input_physical_views = MultiBatchVmapTransform::logicalToPhysical( in batchedTensorInplaceForLoopFallback()313 const auto input_physical_views = MultiBatchVmapTransform::logicalToPhysical( in batchedTensorForLoopFallback()
193 ASSERT_THROW(MultiBatchVmapTransform::logicalToPhysical(tensor), c10::Error); in TEST()201 auto result = MultiBatchVmapTransform::logicalToPhysical(batched); in TEST()210 auto result = MultiBatchVmapTransform::logicalToPhysical(batched); in TEST()220 auto result = MultiBatchVmapTransform::logicalToPhysical(batched); in TEST()238 auto result = MultiBatchVmapTransform::logicalToPhysical(batched); in TEST()270 auto result = MultiBatchVmapTransform::logicalToPhysical(batched); in TEST()396 auto outputs = BroadcastingVmapTransform::logicalToPhysical(inputs); in checkBroadcastingVmapTransform()880 auto outputs = MultiBatchVmapTransform::logicalToPhysical(inputs); in checkMultiBatchVmapTransform()