Home
last modified time | relevance | path

Searched refs:FilterDimensions (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.cc850 return FilterDimensions( in DropDegenerateDimensions()
1444 /* static */ Shape ShapeUtil::FilterDimensions( in FilterDimensions() function in xla::ShapeUtil
Dshape_util.h643 static Shape FilterDimensions(const std::function<bool(int64)>& p,
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_creation_utils.cc435 ShapeUtil::FilterDimensions( in ElideDegenerateDims()
Dalgebraic_simplifier.cc422 Shape shape = ShapeUtil::FilterDimensions( in AddReduce()
3776 Shape reduce_shape = ShapeUtil::FilterDimensions( in HandleReduceWindow()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc2240 Shape reduction_kept_element_shape = ShapeUtil::FilterDimensions( in EmitEpilogueForReduction()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc2215 const Shape logical_element_shape = ShapeUtil::FilterDimensions( in HandleSlice()