Searched refs:TrueRank (Results 1 – 5 of 5) sorted by relevance
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()
207 output_rank = std::max(output_rank, ShapeUtil::TrueRank(subshape)); in EffectivelyAtMostUnary()220 return ShapeUtil::TrueRank(operand->shape()) >= output_rank; in EffectivelyAtMostUnary()
1070 if (ShapeUtil::TrueRank(operand->shape()) == 1 && in ChooseOperandLayoutFromOutputLayout()1071 ShapeUtil::TrueRank(instruction->shape()) == 1) { in ChooseOperandLayoutFromOutputLayout()1132 if (ShapeUtil::TrueRank(operand->shape()) == 1 && in ChooseOutputLayoutFromOperandLayout()1133 ShapeUtil::TrueRank(user->shape()) == 1) { in ChooseOutputLayoutFromOperandLayout()
348 static int64 TrueRank(const Shape& shape);360 return shape.IsArray() && TrueRank(shape) == 0; in IsEffectiveScalar()
195 /* static */ int64 ShapeUtil::TrueRank(const Shape& shape) { in TrueRank() function in xla::ShapeUtil