Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.cc825 std::vector<int64> source_indices(indices.begin(), indices.end()); in ExchangeHalo() local
826 source_indices[dim] -= i + 1; in ExchangeHalo()
828 target.tile_assignment()(source_indices), device); in ExchangeHalo()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.cc512 DimensionVector source_indices(rank, 0); in HandleConcatenate() local
518 GetEvaluatedLiteralFor(operand), source_indices, dest_indices, in HandleConcatenate()