Searched refs:first_slice_operand_shape (Results 1 – 1 of 1) sorted by relevance
504 absl::optional<Shape> first_slice_operand_shape; in GetConsistentInputShapeForRootSlices() local511 if (first_slice_operand_shape.has_value()) { in GetConsistentInputShapeForRootSlices()513 if (!ShapeUtil::EqualIgnoringElementType(*first_slice_operand_shape, in GetConsistentInputShapeForRootSlices()520 first_slice_operand_shape = TypeToShape(slice.operand().getType()); in GetConsistentInputShapeForRootSlices()523 if (!first_slice_operand_shape.has_value()) { in GetConsistentInputShapeForRootSlices()526 return *first_slice_operand_shape; in GetConsistentInputShapeForRootSlices()