Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc545 auto* scatter_instr = xla::Cast<xla::HloScatterInstruction>(instr); in GetScatterDimensionNumbers() local
548 scatter_instr->scatter_dimension_numbers(); in GetScatterDimensionNumbers()
564 auto* scatter_instr = xla::Cast<xla::HloScatterInstruction>(instr); in EmitScatterOp() local
570 builder_.getBoolAttr(scatter_instr->indices_are_sorted())); in EmitScatterOp()
572 builder_.getBoolAttr(scatter_instr->unique_indices())); in EmitScatterOp()
576 *scatter_instr->called_computations()[0], &scatter.update_computation(), in EmitScatterOp()