Searched refs:expected_updates_rank (Results 1 – 2 of 2) sorted by relevance
155 int64 expected_updates_rank = expected_updates_dims.size(); in XlaScatter() local156 if (updates_rank == 0 && expected_updates_rank != 0) { in XlaScatter()
3204 int64 expected_updates_rank = expanded_scatter_indices_shape.size() - 1 + in InferScatterShape() local3206 if (updates_shape.rank() != expected_updates_rank) { in InferScatterShape()3208 expected_updates_rank, updates_shape.rank()); in InferScatterShape()