HomeSort by relevance Sort by last modified time
    Searched refs:host_call (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/python/tpu/
tpu_estimator.py 292 `host_call` is a tuple of a `function` and a list or dictionary of `tensors`
293 to pass to that function and returns a list of Tensors. `host_call` currently
298 dimension, and so must be >= rank 1. The `host_call` is useful for writing
310 host_call=None,
318 if host_call is not None:
319 host_calls['host_call'] = host_call
340 host_call=host_call,
350 if self.host_call is not None
    [all...]
  /external/tensorflow/tensorflow/contrib/eager/python/examples/revnet/
main_estimator_tpu.py 43 model to the `metric_fn`, provide as part of the `host_call`. See
48 element in the tuple passed to `host_call`.
147 # summary op needs to be run on the host CPU via host_call. host_call
154 host_call = (_host_call_fn, [gs_t, loss_t, lr_t])
157 mode=mode, loss=loss, train_op=train_op, host_call=host_call)
384 help=("Skip the host_call which is executed every training step. This is"
387 " be a performance drop if host_call function is slow and cannot"

Completed in 74 milliseconds