Searched refs:ReshardAsWindowedInput (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 546 PartitionedHlo::ReshardAsWindowedInput(const Window& window, in ReshardAsWindowedInput() function in xla::spmd::PartitionedHlo
769 return Reshard(target).ReshardAsWindowedInput(window, target, pad_value); in ReshardAsWindowedInput()
814 return Replicate().ReshardAsWindowedInput(window, target, pad_value); in ReshardAsWindowedInput()
1601 auto reshard_operand = operand.ReshardAsWindowedInput( in HandleSlice()
2071 auto reshard_operand = operand.ReshardAsWindowedInput( in HandleReshape()
2126 auto reshard_output = tmp_output.ReshardAsWindowedInput( in HandleReshape()
2645 lhs.ReshardAsWindowedInput(window, hlo->sharding(), replicated_rhs, in HandlePad()
3115 auto resharded_operand_and_window = operand.ReshardAsWindowedInput( in HandleReduceWindow()
|
| D | spmd_partitioner.h | 330 absl::optional<WindowedInputShardReturnValue> ReshardAsWindowedInput(
|
| D | convolution_handler.cc | 787 lhs.ReshardAsWindowedInput(operand_window, target_operand_sharding, zero); in PartitionConvolutionTiledOutput()
|
| D | spmd_partitioner_util.cc | 1097 auto reshard_window = original.ReshardAsWindowedInput( in HaloExchangeToPadOnLeft()
|