Searched refs:halo_start_indices (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_util.cc | 837 std::vector<int64> halo_start_indices(halo_shape.rank(), 0); in ExchangeHalo() local
838 halo_start_indices[dim] = hlo->shape().dimensions(dim) - halo_size; in ExchangeHalo()
841 halo_shape, hlo, halo_start_indices, hlo->shape().dimensions(), in ExchangeHalo()
871 std::vector<int64> halo_start_indices(halo_shape.rank(), 0); in ExchangeHalo() local
874 halo_shape, hlo, halo_start_indices, halo_shape.dimensions(), in ExchangeHalo()
|