mlir::CallInterfaceCallable callee) {
  return SetPartitionCalleeFromCallable(*this, callee);
}
void StatefulPartitionedCallOp::setCalleeFromCallable(
    CallInterfaceCallable callee) {
  return SetPartitionCalleeFromCallable(*this, callee);
}
void TPUPartitionedCallOp::setCalleeFromCallable(
    mlir::CallInterfaceCallable callee) {
  return SetPartitionCalleeFromCallable(*this, callee);
}