def METHOD_NAME(
    self,
    name: Union[str, bytes, core.BlobReference],
    arr: Union[caffe2_pb2.TensorProto, np.ndarray],
    device_option: Optional[caffe2_pb2.DeviceOption] = None,
) -> bool: ...