Searched refs:num_scatter_dims (Results 1 – 1 of 1) sorted by relevance
117 int64 num_scatter_dims = scatter_indices_shape.dimensions_size(); in AdjustScatterDims() local119 --num_scatter_dims; in AdjustScatterDims()121 if (num_scatter_dims == 0) { in AdjustScatterDims()127 return CollapseFirstNDims(updates, num_scatter_dims); in AdjustScatterDims()