Searched defs:physical_args (Results 1 – 1 of 1) sorted by relevance
99 auto physical_args = BroadcastingVmapTransform::logicalToPhysical({self, other}); in binary_pointwise_batching_rule() local149 auto physical_args = BroadcastingVmapTransform::logicalToPhysical( in binary_pointwise_batching_rule() local257 auto physical_args = in fill_inplace_tensor_batching_rule() local836 auto physical_args = MultiBatchVmapTransform::logicalToPhysical({self, other}); in mv_batching_rule() local883 auto physical_args = MultiBatchVmapTransform::logicalToPhysical({self, other}); in dot_batching_rule() local898 auto physical_args = BroadcastingVmapTransform::logicalToPhysical({self, other}); in bmm_batching_rule() local924 auto physical_args = MultiBatchVmapTransform::logicalToPhysical({self, other}); in mm_batching_rule() local1070 auto physical_args = BroadcastingVmapTransform::logicalToPhysical({self, other}); in comparison_pointwise_batching_rule() local