Searched refs:TrueRank (Results 1 – 4 of 4) sorted by relevance
178 output_rank = std::max(output_rank, ShapeUtil::TrueRank(subshape)); in EffectivelyAtMostUnary()191 return ShapeUtil::TrueRank(operand->shape()) >= output_rank; in EffectivelyAtMostUnary()
1013 if (ShapeUtil::TrueRank(operand->shape()) == 1 && in ChooseOperandLayoutFromOutputLayout()1014 ShapeUtil::TrueRank(instruction->shape()) == 1) { in ChooseOperandLayoutFromOutputLayout()1074 if (ShapeUtil::TrueRank(operand->shape()) == 1 && in ChooseOutputLayoutFromOperandLayout()1075 ShapeUtil::TrueRank(user->shape()) == 1) { in ChooseOutputLayoutFromOperandLayout()
303 static int64 TrueRank(const Shape& shape);315 return shape.IsArray() && TrueRank(shape) == 0; in IsEffectiveScalar()
137 /* static */ int64 ShapeUtil::TrueRank(const Shape& shape) { in TrueRank() function in xla::ShapeUtil