Searched refs:_to_replicate_tensor (Results 1 – 3 of 3) sorted by relevance
296 full_tensor = shard_placement._to_replicate_tensor(314 full_tensor = shard_placement._to_replicate_tensor(354 full_tensor = shard_placement_dim1._to_replicate_tensor(363 full_tensor = shard_placement_dim0._to_replicate_tensor(399 full_tensor = shard_placement_dim1._to_replicate_tensor(408 full_tensor = shard_placement_dim0._to_replicate_tensor(457 full_tensor = shard_placement_dim1._to_replicate_tensor(466 full_tensor = shard_placement_dim0._to_replicate_tensor(
200 new_local_tensor = current_placement._to_replicate_tensor(256 new_local_tensor = current_placement._to_replicate_tensor(
218 def _to_replicate_tensor( member in Shard484 def _to_replicate_tensor( member in _StridedShard