Searched refs:ShapesSame (Results 1 – 2 of 2) sorted by relevance
238 if (!ShapesSame(operand->shape(), parameter->shape())) { in CheckOperandAndParameter()262 if (!ShapesSame(outfeed->outfeed_shape(), outfeed->operand(0)->shape())) { in HandleOutfeed()346 if (!ShapesSame(compare_shape, ShapeUtil::MakeShape(PRED, {}))) { in HandleSort()524 if (!ShapesSame(fused_param->shape(), fusion->operand(param_no)->shape())) { in HandleFusion()657 if (!ShapesSame(conditional_shape, ShapeUtil::MakeShape(PRED, {}))) { in HandleWhile()881 return ShapesSame(instruction->shape(), inferred_shape); in CheckShape()
123 bool ShapesSame(const Shape& a, const Shape& b) { in ShapesSame() function