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)