Searched refs:TrueRank (Results 1 – 4 of 4) sorted by relevance
199 output_rank = std::max(output_rank, ShapeUtil::TrueRank(subshape)); in EffectivelyAtMostUnary()212 return ShapeUtil::TrueRank(operand->shape()) >= output_rank; in EffectivelyAtMostUnary()
1069 if (ShapeUtil::TrueRank(operand->shape()) == 1 && in ChooseOperandLayoutFromOutputLayout()1070 ShapeUtil::TrueRank(instruction->shape()) == 1) { in ChooseOperandLayoutFromOutputLayout()1131 if (ShapeUtil::TrueRank(operand->shape()) == 1 && in ChooseOutputLayoutFromOperandLayout()1132 ShapeUtil::TrueRank(user->shape()) == 1) { in ChooseOutputLayoutFromOperandLayout()
305 static int64 TrueRank(const Shape& shape);317 return shape.IsArray() && TrueRank(shape) == 0; in IsEffectiveScalar()
153 /* static */ int64 ShapeUtil::TrueRank(const Shape& shape) { in TrueRank() function in xla::ShapeUtil