| 127 | class IfOp : public AsyncOpKernel { |
| 128 | public: |
| 129 | explicit IfOp(OpKernelConstruction* ctx) : AsyncOpKernel(ctx) { |
| 130 | auto lib = ctx->function_library(); |
| 131 | OP_REQUIRES(ctx, lib != nullptr, errors::Internal("No function library")); |
| 132 | OP_REQUIRES_OK(ctx, ctx->GetAttr("then_branch", &then_func_)); |
| 133 | OP_REQUIRES_OK(ctx, ctx->GetAttr("else_branch", &else_func_)); |
| 134 | } |
| 135 | |
| 136 | ~IfOp() override {} |
| 137 |
nothing calls this directly
no test coverage detected