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::__anon23a946722e11::UpdateScatterIndexToInputIndex