Searched refs:scatter_dims_seen (Results 1 – 1 of 1) sorted by relevance
3613 int64 scatter_dims_seen = 0; in InferScatterShape() local3620 if (scatter_dims_seen == scatter_dim_numbers.index_vector_dim()) { in InferScatterShape()3621 ++scatter_dims_seen; in InferScatterShape()3624 expanded_scatter_indices_shape[scatter_dims_seen]) { in InferScatterShape()3631 expanded_scatter_indices_shape[scatter_dims_seen]); in InferScatterShape()3633 ++scatter_dims_seen; in InferScatterShape()