llvm::ArrayRef<int64_t> shape_ref =
        mlir::cast<TensorType>(tensor_attr.getType()).getShape();
    if (mlir::failed(check_shape(shape_ref))) return std::nullopt;

    shape = std::vector<int32_t>(shape_ref.begin(), shape_ref.end());
  } else if (type.hasRank()) {
    llvm::ArrayRef<int64_t> shape_ref = type.getShape();
    if (mlir::failed(check_shape(shape_ref))) return std::nullopt;

      return ComputeOutputComponent(ValuePort(id.getInput()), values);
    return nullptr;
  }

  if (auto shape_op = dyn_cast<TF::ShapeOp>(op)) {
    // No shape available in an unranked tensor type.
    auto operand_ty =
        mlir::dyn_cast<RankedTensorType>(shape_op.getOperand().getType());
    if (!operand_ty) return nullptr;

    // Shape op has a single output so the first element should always be zero