Searched refs:scatter_dim_to_operand_dim (Results 1 – 2 of 2) sorted by relevance
3511 int64 scatter_dim_to_operand_dim = in ValidateScatterDimensionNumbers() local3513 if (scatter_dim_to_operand_dim < 0 || in ValidateScatterDimensionNumbers()3514 scatter_dim_to_operand_dim >= operand_shape.dimensions_size()) { in ValidateScatterDimensionNumbers()3518 operand_shape.dimensions_size(), i, scatter_dim_to_operand_dim); in ValidateScatterDimensionNumbers()
2867 for (int64 scatter_dim_to_operand_dim : scatter_dims_to_operand_dims) { in MakeScatterDimNumbers() local2869 scatter_dim_to_operand_dim); in MakeScatterDimNumbers()