(self, step, *args)
Function that will run on the host machine.
Function that will run on the host machine.
Function that will run on the host machine .
def _host_call_fn(self, step, *args):
    """Function that will run on the host machine."""
    # Host call receives values from all tensor cores (concatenate on the
    # batch dimension). Step is the same for all cores.
    step = step[0]
    logging.info("host_call_fn: args=%s", args)
    with summary.create_file_writer(self._log_dir).as_default():
      with summary.record_summaries_every_n_global_steps(
          self._save_summary_steps, step):
        for i, e in enumerate(self._entries):
          value = e.reduce_fn(args[i])
          e.summary_fn(e.name, value, step=step)
        return summary.all_summary_ops()
def _host_call_fn ( self , step , * args ) : # Host call receives values from all tensor cores (concatenate on the # batch dimension). Step is the same for all cores. step = step [ 0 ] logging . info ( "host_call_fn: args=%s" , args ) with summary . create_file_writer ( self . _log_dir ) . as_default ( ) : with summary . record_summaries_every_n_global_steps ( self . _save_summary_steps , step ) : for i , e in enumerate ( self . _entries ) : value = e . reduce_fn ( args [ i ] ) e . summary_fn ( e . name , value , step = step ) return summary . all_summary_ops ( )