Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhorizontal_loop_fusion.cc153 constexpr int64 kShapeThreshold = 128 * 2048; in IsProfitableFusionCandidate() local
162 if (ShapeUtil::ElementsIn(shape) > kShapeThreshold) { in IsProfitableFusionCandidate()
167 if (ShapeUtil::ElementsIn(shape) > kShapeThreshold) { in IsProfitableFusionCandidate()