Searched refs:scatter_instr (Results 1 – 1 of 1) sorted by relevance
545 auto* scatter_instr = xla::Cast<xla::HloScatterInstruction>(instr); in GetScatterDimensionNumbers() local548 scatter_instr->scatter_dimension_numbers(); in GetScatterDimensionNumbers()564 auto* scatter_instr = xla::Cast<xla::HloScatterInstruction>(instr); in EmitScatterOp() local570 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()