Searched refs:expected_updates_dims (Results 1 – 1 of 1) sorted by relevance
150 std::vector<int64> expected_updates_dims(indices_dims.begin(), in XlaScatter() local153 expected_updates_dims.push_back(buffer_shape.dimensions(dim)); in XlaScatter()155 int64 expected_updates_rank = expected_updates_dims.size(); in XlaScatter()157 new_updates = xla::Broadcast(updates, expected_updates_dims); in XlaScatter()