Searched refs:update_window_dim (Results 1 – 2 of 2) sorted by relevance
3226 auto update_window_dim = scatter_dim_numbers.update_window_dims(i); in InferScatterShape() local3227 if (updates_shape.dimensions(update_window_dim) > in InferScatterShape()3233 update_window_dim, updates_shape.dimensions(update_window_dim), in InferScatterShape()
2435 for (int64 update_window_dim : update_window_dims) { in MakeScatterDimNumbers() local2436 scatter_dim_numbers.add_update_window_dims(update_window_dim); in MakeScatterDimNumbers()