Searched defs:compile_op (Results 1 – 2 of 2) sorted by relevance
399 auto compile_op = builder->create<TF::_TPUCompileMlirOp>( in BuildCompileOp() local472 Operation* compile_op, tf_device::ClusterFuncOp cluster_func, in BuildParallelExecuteOp()558 void BuildTPUCompileSucceededAssertOp(Operation* compile_op, in BuildTPUCompileSucceededAssertOp()629 Operation* compile_op = BuildCompileOp( in Rewrite() local
276 auto compile_op = [&](XlaCompiler* compiler, in CompileSingleOp() local