Searched defs:gather_dimension_numbers (Results 1 – 4 of 4) sorted by relevance
157 xla.GatherDimensionNumbers gather_dimension_numbers = 33; field
1659 const GatherDimensionNumbers& gather_dimension_numbers() const { in gather_dimension_numbers() function
668 auto gather_dimension_numbers = absl::make_unique<GatherDimensionNumbers>( in CreateFromProto() local4236 const GatherDimensionNumbers& HloInstruction::gather_dimension_numbers() const { in gather_dimension_numbers() function in xla::HloInstruction
2737 const GatherDimensionNumbers& gather_dimension_numbers) { in GatherDimensionNumbersToString()