Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc1608 TF_RET_CHECK(!reshard_operand->dynamic_slice_index_on_output.has_value()); in HandleSlice()
2078 TF_RET_CHECK(!reshard_operand->dynamic_slice_index_on_output.has_value()); in HandleReshape()
2133 TF_RET_CHECK(!reshard_output->dynamic_slice_index_on_output.has_value()); in HandleReshape()
2675 if (!reshard_operand->dynamic_slice_index_on_output) { in HandlePad()
2681 *reshard_operand->dynamic_slice_index_on_output, in HandlePad()
3134 if (!resharded_operand_and_window->dynamic_slice_index_on_output in HandleReduceWindow()
3141 *resharded_operand_and_window->dynamic_slice_index_on_output, in HandleReduceWindow()
Dspmd_partitioner.h266 absl::optional<std::vector<HloInstruction*>> dynamic_slice_index_on_output; member
Dconvolution_handler.cc804 if (!resharded_operand_and_window->dynamic_slice_index_on_output in PartitionConvolutionTiledOutput()
811 *resharded_operand_and_window->dynamic_slice_index_on_output, in PartitionConvolutionTiledOutput()
Dspmd_partitioner_util.cc1105 CHECK(!reshard_window->dynamic_slice_index_on_output.has_value()); in HaloExchangeToPadOnLeft()