Searched refs:device_return (Results 1 – 2 of 2) sorted by relevance
348 auto device_return = mlir::cast<mlir::tf_device::ReturnOp>(owner); in TraceUseToNextTFOp() local350 device_return->getParentOfType<mlir::tf_device::ClusterOp>(); in TraceUseToNextTFOp()
2056 auto device_return = in CreateCompiledModule() local2058 OpBuilder builder(device_return.getOperation()); in CreateCompiledModule()2059 builder.create<func::ReturnOp>(device_return.getLoc(), in CreateCompiledModule()2060 device_return.getOperands()); in CreateCompiledModule()2061 device_return.erase(); in CreateCompiledModule()