Home
last modified time | relevance | path

Searched refs:xla_scatter_dim (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc547 const xla::ScatterDimensionNumbers& xla_scatter_dim = in GetScatterDimensionNumbers() local
550 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()