Searched refs:start_indices_halo (Results 1 – 1 of 1) sorted by relevance
410 std::vector<int64> start_indices_halo(rank, 0), in HaloDuplicateWithSlice() local414 start_indices_halo[activations_batch_dim] = 1; in HaloDuplicateWithSlice()418 MakeSliceHlo(activations, start_indices_halo, in HaloDuplicateWithSlice()