Searched refs:host_op (Results 1 – 2 of 2) sorted by relevance
61 void WrapOpInLaunch(Operation* host_op, llvm::StringRef host_device) { in WrapOpInLaunch() argument62 OpBuilder builder(host_op); in WrapOpInLaunch()65 host_op->getLoc(), builder.getStringAttr(host_device), in WrapOpInLaunch()66 /*result_types=*/host_op->getResultTypes()); in WrapOpInLaunch()67 host_op->replaceAllUsesWith(launch_op); in WrapOpInLaunch()73 .create<tf_device::ReturnOp>(host_op->getLoc(), host_op->getResults()) in WrapOpInLaunch()76 host_op->removeAttr(Identifier::get(kXlaOutsideCompilationAttr, context)); in WrapOpInLaunch()77 host_op->removeAttr(Identifier::get(kDeviceAttr, context)); in WrapOpInLaunch()78 host_op->moveBefore(return_op); in WrapOpInLaunch()
236 Operation* host_op = outside_compiled_ops.front(); in MarkVariantInputsOutputs() local238 host_op->walk([&](Operation* op) { in MarkVariantInputsOutputs()