Searched refs:scatter_dim_to_operand_dim (Results 1 – 2 of 2) sorted by relevance
3136 int64 scatter_dim_to_operand_dim = in ValidateScatterDimensionNumbers() local3138 if (scatter_dim_to_operand_dim < 0 || in ValidateScatterDimensionNumbers()3139 scatter_dim_to_operand_dim >= operand_shape.dimensions_size()) { in ValidateScatterDimensionNumbers()3143 operand_shape.dimensions_size(), i, scatter_dim_to_operand_dim); in ValidateScatterDimensionNumbers()
2441 for (int64 scatter_dim_to_operand_dim : scatter_dims_to_operand_dims) { in MakeScatterDimNumbers() local2443 scatter_dim_to_operand_dim); in MakeScatterDimNumbers()