| 749 | } |
| 750 | |
| 751 | Status CheckResultType(const Datum& out, const char* function_name) override { |
| 752 | const auto& type = out.type(); |
| 753 | if (type != nullptr && !type->Equals(*output_type_.type)) { |
| 754 | return Status::TypeError( |
| 755 | "kernel type result mismatch for function '", function_name, "': declared as ", |
| 756 | output_type_.type->ToString(), ", actual is ", type->ToString()); |
| 757 | } |
| 758 | return Status::OK(); |
| 759 | } |
| 760 | |
| 761 | ExecContext* exec_context() { return kernel_ctx_->exec_context(); } |
| 762 | KernelState* state() { return kernel_ctx_->state(); } |