Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/
H A Dxla_sharding.py205 def apply_to_tensor(self, member in Sharding
309 return Sharding.replicate().apply_to_tensor(
320 return Sharding.assign_device(device).apply_to_tensor(
341 return Sharding.tile(tile_assignment).apply_to_tensor(
365 input_shape).apply_to_tensor(
386 return Sharding.partial_tile(tile_assignment).apply_to_tensor(
584 return sharding.apply_to_tensor(