Searched refs:compile_op (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compilation_cache.cc | 212 auto compile_op = [&](XlaCompiler* compiler, in CompileSingleOp() local
221 return CompileImpl(options, name, args, compile_op, in CompileSingleOp()
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_estimator.py | 2838 compile_op, loss, host_call, scaffold, training_hooks = (
2891 tpu_compile_op=compile_op,
2951 compile_op, total_loss, host_calls, scaffold, eval_hooks = (
3012 tpu_compile_op=compile_op,
3034 (compile_op, dummy_predict_op, host_calls,
3091 tpu_compile_op=compile_op,
3180 (compile_op, loss,) = tpu.split_compile_and_shard(
3189 return compile_op, loss, host_calls, scaffold, captured_eval_hooks.get()
3205 (compile_op, loss,) = tpu.split_compile_and_shard(
3214 return compile_op, loss, host_call, scaffold, captured_training_hooks.get()
[all …]
|
| D | tpu.py | 1181 compile_op, outputs = split_compile_and_replicate(
1198 return compile_op, [outputs[0]]
1232 return compile_op, results
|
| /external/tensorflow/tensorflow/contrib/tpu/python/tpu/ |
| D | keras_support.py | 1095 compile_op, execute_op = tpu.split_compile_and_replicate(
1115 compile_op,
1126 result = K.get_session().run(tpu_model_ops.compile_op)
|