Searched defs:model_fn_wrapper (Results 1 – 1 of 1) sorted by relevance
3168 def _eval_on_tpu_system(ctx, model_fn_wrapper, dequeue_fn): argument3192 def _train_on_tpu_system(ctx, model_fn_wrapper, dequeue_fn): argument3217 def _predict_on_tpu_system(ctx, model_fn_wrapper, dequeue_fn): argument