Searched refs:expanded_scatter_indices_shape (Results 1 – 1 of 1) sorted by relevance
3687 std::vector<int64_t> expanded_scatter_indices_shape = in InferScatterShape() local3689 if (expanded_scatter_indices_shape.size() == in InferScatterShape()3691 expanded_scatter_indices_shape.push_back(1); in InferScatterShape()3718 expanded_scatter_indices_shape.size() - 1 + in InferScatterShape()3726 operand_shape, expanded_scatter_indices_shape, updates_shape, in InferScatterShape()3753 expanded_scatter_indices_shape[scatter_dims_seen]) { in InferScatterShape()3760 expanded_scatter_indices_shape[scatter_dims_seen]); in InferScatterShape()