Searched refs:compile_op (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_rewrite_pass.cc | 330 Operation* BuildExecuteOp(Operation* compile_op, in BuildExecuteOp() argument
336 tensor_inputs.push_back(compile_op->getResult(1)); in BuildExecuteOp()
349 void BuildTPUCompileSucceededAssertOp(Operation* compile_op, in BuildTPUCompileSucceededAssertOp() argument
351 OperationState assert_op_state(compile_op->getLoc(), in BuildTPUCompileSucceededAssertOp()
353 assert_op_state.addOperands(compile_op->getResult(0)); in BuildTPUCompileSucceededAssertOp()
448 Operation* compile_op = in Rewrite() local
451 if (!compile_op) return failure(); in Rewrite()
459 compile_result_op.output().replaceAllUsesWith(compile_op->getResult(0)); in Rewrite()
461 BuildTPUCompileSucceededAssertOp(compile_op, builder); in Rewrite()
464 Operation* execute_op = BuildExecuteOp(compile_op, launch_func, builder); in Rewrite()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compilation_cache.cc | 219 auto compile_op = [&](XlaCompiler* compiler, in CompileSingleOp() local
228 return CompileImpl(options, name, args, compile_op, in CompileSingleOp()
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu.py | 1487 compile_op, outputs = split_compile_and_replicate(
1504 return compile_op, [outputs[0]]
1538 return compile_op, results
|