Searched refs:xla_scatter_dim (Results 1 – 1 of 1) sorted by relevance
547 const xla::ScatterDimensionNumbers& xla_scatter_dim = in GetScatterDimensionNumbers() local550 GetI64DenseElementsAttr(xla_scatter_dim.update_window_dims()), in GetScatterDimensionNumbers()551 GetI64DenseElementsAttr(xla_scatter_dim.inserted_window_dims()), in GetScatterDimensionNumbers()552 GetI64DenseElementsAttr(xla_scatter_dim.scatter_dims_to_operand_dims()), in GetScatterDimensionNumbers()553 builder_.getI64IntegerAttr(xla_scatter_dim.index_vector_dim()), in GetScatterDimensionNumbers()