Searched defs:scatter_dimension_numbers (Results 1 – 6 of 6) sorted by relevance
249 xla.ScatterDimensionNumbers scatter_dimension_numbers = 48; field
2156 const ScatterDimensionNumbers& scatter_dimension_numbers() const { in scatter_dimension_numbers() function
860 auto scatter_dimension_numbers = in CreateFromProto() local4826 const ScatterDimensionNumbers& HloInstruction::scatter_dimension_numbers() in scatter_dimension_numbers() function in xla::HloInstruction
3158 const ScatterDimensionNumbers& scatter_dimension_numbers) { in ScatterDimensionNumbersToString()
625 auto scatter_dimension_numbers = mhlo::ScatterDimensionNumbersAttr::get( in GetScatterDimensionNumbers() local
3157 auto scatter_dimension_numbers = scatter_op.scatter_dimension_numbers(); in matchAndRewrite() local