Searched refs:TrueRank (Results 1 – 4 of 4) sorted by relevance
140 output_rank = std::max(output_rank, ShapeUtil::TrueRank(subshape)); in EffectivelyUnary()152 return ShapeUtil::TrueRank(operand->shape()) >= in EffectivelyUnary()
918 if (ShapeUtil::TrueRank(operand->shape()) == 1 && in ChooseOperandLayoutFromOutputLayout()991 ShapeUtil::TrueRank(user->shape()) == 1) { in ChooseOutputLayoutFromOperandLayout()
268 static int64 TrueRank(const Shape& shape);280 return !IsTuple(shape) && !IsOpaque(shape) && TrueRank(shape) == 0; in IsEffectiveScalar()
161 /* static */ int64 ShapeUtil::TrueRank(const Shape& shape) { in TrueRank() function in xla::ShapeUtil