Searched defs:scatter_dimension_numbers (Results 1 – 5 of 5) sorted by relevance
194 xla.ScatterDimensionNumbers scatter_dimension_numbers = 48; field
1706 const ScatterDimensionNumbers& scatter_dimension_numbers() const { in scatter_dimension_numbers() function
685 auto scatter_dimension_numbers = in CreateFromProto() local4244 const ScatterDimensionNumbers& HloInstruction::scatter_dimension_numbers() in scatter_dimension_numbers() function in xla::HloInstruction
2834 const ScatterDimensionNumbers& scatter_dimension_numbers) { in ScatterDimensionNumbersToString()
549 auto scatter_dimension_numbers = mhlo::ScatterDimensionNumbers::get( in GetScatterDimensionNumbers() local