Searched defs:self_physical (Results 1 – 2 of 2) sorted by relevance
78 auto self_physical = MultiBatchVmapTransform::logicalToPhysical(self); in sum_batching_rule() local109 auto self_physical = MultiBatchVmapTransform::logicalToPhysical(self); in binary_pointwise_batching_rule() local156 auto self_physical = MultiBatchVmapTransform::logicalToPhysical(self); in expand_batching_rule() local194 auto self_physical = MultiBatchVmapTransform::logicalToPhysical(self); in chunk_batching_rule() local202 auto self_physical = MultiBatchVmapTransform::logicalToPhysical(self); in clamp_batching_rule() local208 auto self_physical = MultiBatchVmapTransform::logicalToPhysical(self); in clamp_min_batching_rule() local214 auto self_physical = MultiBatchVmapTransform::logicalToPhysical(self); in clamp_max_batching_rule() local220 auto self_physical = MultiBatchVmapTransform::logicalToPhysical(self); in tensor_split_sections_batching_rule() local228 auto self_physical = MultiBatchVmapTransform::logicalToPhysical(self); in tensor_split_indices_batching_rule() local236 auto self_physical = MultiBatchVmapTransform::logicalToPhysical(self); in unsqueeze_batching_rule() local[all …]
242 auto self_physical = MultiBatchVmapTransform::logicalToPhysical(self); in split_batching_rule() local254 auto self_physical = MultiBatchVmapTransform::logicalToPhysical(self); in split_with_sizes_batching_rule() local266 auto self_physical = MultiBatchVmapTransform::logicalToPhysical(self); in split_with_sizes_copy_batching_rule() local278 auto self_physical = MultiBatchVmapTransform::logicalToPhysical(self); in unbind_batching_rule() local