| 2464 | } |
| 2465 | |
| 2466 | TF_Buffer* TF_GetRegisteredKernelsForOp(const char* name, TF_Status* status) { |
| 2467 | tensorflow::KernelList kernel_list = |
| 2468 | tensorflow::GetRegisteredKernelsForOp(name); |
| 2469 | TF_Buffer* ret = TF_NewBuffer(); |
| 2470 | status->status = MessageToBuffer(kernel_list, ret); |
| 2471 | if (TF_GetCode(status) != TF_OK) { |
| 2472 | TF_DeleteBuffer(ret); |
| 2473 | return nullptr; |
| 2474 | } |
| 2475 | return ret; |
| 2476 | } |
| 2477 | |
| 2478 | // TF_Server functions ---------------------------------------------- |
| 2479 |