Searched refs:scatter_dims_seen (Results 1 – 1 of 1) sorted by relevance
3238 int64 scatter_dims_seen = 0; in InferScatterShape() local3245 if (scatter_dims_seen == scatter_dim_numbers.index_vector_dim()) { in InferScatterShape()3246 ++scatter_dims_seen; in InferScatterShape()3249 expanded_scatter_indices_shape[scatter_dims_seen]) { in InferScatterShape()3256 expanded_scatter_indices_shape[scatter_dims_seen]); in InferScatterShape()3258 ++scatter_dims_seen; in InferScatterShape()