Searched refs:ElideDegenerateDims (Results 1 – 5 of 5) sorted by relevance
68 ElideDegenerateDims(lhs, degenerate_dims)); in ElideDegenerateBatchDimensionFromBatchDot()70 ElideDegenerateDims(rhs, degenerate_dims)); in ElideDegenerateBatchDimensionFromBatchDot()
102 return ElideDegenerateDims(accumulator, {0}); in AdjustBatchDimsInAccumulator()192 ElideDegenerateDims(index_vector_2d, {0})); in GatherLoopBody()206 ElideDegenerateDims(gathered_slice, in GatherLoopBody()
254 ElideDegenerateDims(index_vector_2d, {0})); in ScatterLoopBody()275 ElideDegenerateDims(update_slice, {0})); in ScatterLoopBody()
261 StatusOr<HloInstruction*> ElideDegenerateDims(
491 StatusOr<HloInstruction*> ElideDegenerateDims( in ElideDegenerateDims() function