Searched defs:dim_numbers_ (Results 1 – 1 of 1) sorted by relevance
2362 const GatherDimensionNumbers& dim_numbers_; member in xla::OutputBatchIndexToInputIndex2751 const ScatterDimensionNumbers& dim_numbers_; member in xla::__anon9f8d6ac12e11::UpdateScatterIndexToInputIndex