Home
last modified time | relevance | path

Searched defs:gather_dimension_numbers (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo.proto212 xla.GatherDimensionNumbers gather_dimension_numbers = 33; field
H A Dhlo_instructions.h2106 const GatherDimensionNumbers& gather_dimension_numbers() const { in gather_dimension_numbers() function
H A Dhlo_instruction.cc841 auto gather_dimension_numbers = std::make_unique<GatherDimensionNumbers>( in CreateFromProto() local
4818 const GatherDimensionNumbers& HloInstruction::gather_dimension_numbers() const { in gather_dimension_numbers() function in xla::HloInstruction
H A Dhlo_instructions.cc3061 const GatherDimensionNumbers& gather_dimension_numbers) { in GatherDimensionNumbersToString()