Searched defs:compile_op (Results 1 – 3 of 3) sorted by relevance
89 mlir::TF::_TPUCompileMlirOp compile_op, in CreateSendRecvOpsToTransferProgramKey()189 mlir::TF::_TPUCompileMlirOp compile_op, in SendRecvCompilationKey()263 mlir::TF::_TPUCompileMlirOp compile_op = *compile_ops.begin(); in HandleCompilationOps() local
389 auto compile_op = builder->create<TF::_TPUCompileMlirOp>( in BuildCompileOp() local510 Operation* compile_op, tf_device::ClusterFuncOp cluster_func, in AddToParallelExecuteOp()601 void BuildTPUCompileSucceededAssertOp(Operation* compile_op, in BuildTPUCompileSucceededAssertOp()769 Operation* compile_op = in Rewrite() local
83 auto compile_op = ::llvm::dyn_cast<mlir::TF::_TPUCompileMlirOp>( in runOnOperation() local