Home
last modified time | relevance | path

Searched refs:ElideDegenerateDims (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dbatch_dot_simplification.cc50 ElideDegenerateDims(lhs, degenerate_dims)); in ElideDegenerateBatchDimensionFromBatchDot()
52 ElideDegenerateDims(rhs, degenerate_dims)); in ElideDegenerateBatchDimensionFromBatchDot()
Dgather_expander.cc102 return ElideDegenerateDims(accumulator, {0}); in AdjustBatchDimsInAccumulator()
192 ElideDegenerateDims(index_vector_2d, {0})); in GatherLoopBody()
206 ElideDegenerateDims(gathered_slice, in GatherLoopBody()
Dscatter_expander.cc254 ElideDegenerateDims(index_vector_2d, {0})); in ScatterLoopBody()
275 ElideDegenerateDims(update_slice, {0})); in ScatterLoopBody()
Dhlo_creation_utils.h194 StatusOr<HloInstruction*> ElideDegenerateDims(
Dhlo_creation_utils.cc363 StatusOr<HloInstruction*> ElideDegenerateDims( in ElideDegenerateDims() function