def AsyncPush(self, op_name, push_data_cb):
assert self.status_ is SessionStatus.RUNNING
push_job_name = self.inter_user_job_info.input_or_var_op_name2push_job_name[
op_name]
self.LaunchJob(
job_instance_util.MakePushJobInstance(push_job_name, op_name,
push_data_cb))
def _run_push_jobs(self, **kwargs):
for (
input_name,
push_job_name,
) in self.inter_user_job_info_.input_or_var_op_name2push_job_name.items():
if input_name not in kwargs:
raise ValueError('input "{}" is absent'.format(input_name))
input_numpy = kwargs[input_name]
if not isinstance(input_numpy, np.ndarray):
raise ValueError('input "{}" requires numpy.ndarray'.format(input_name))
push_fn = input_blob_util._MakePushNdarrayCallback(input_numpy)
push_job_inst = job_instance_util.MakePushJobInstance(
push_job_name, input_name, push_fn
)
self._run_job(push_job_inst)
