Searched defs:gather_dimension_numbers (Results 1 – 3 of 3) sorted by relevance
153 xla.GatherDimensionNumbers gather_dimension_numbers = 33; field
1366 const GatherDimensionNumbers& gather_dimension_numbers() const { in gather_dimension_numbers() function
525 std::unique_ptr<GatherDimensionNumbers> gather_dimension_numbers = in CreateFromProto() local3516 const GatherDimensionNumbers& HloInstruction::gather_dimension_numbers() const { in gather_dimension_numbers() function in xla::HloInstruction