Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_fusible.cc96 int64 max_true_rank = -1; in LayoutsAreReduceInputFusionFriendly() local
100 ShapeUtil::TrueRank(param->shape()) > max_true_rank) { in LayoutsAreReduceInputFusionFriendly()
101 max_true_rank = ShapeUtil::TrueRank(param->shape()); in LayoutsAreReduceInputFusionFriendly()
107 ShapeUtil::TrueRank(param->shape()) < max_true_rank || in LayoutsAreReduceInputFusionFriendly()