| 2453 | } |
| 2454 | |
| 2455 | TF_Buffer* TF_GetAllRegisteredKernels(TF_Status* status) { |
| 2456 | tensorflow::KernelList kernel_list = tensorflow::GetAllRegisteredKernels(); |
| 2457 | TF_Buffer* ret = TF_NewBuffer(); |
| 2458 | status->status = MessageToBuffer(kernel_list, ret); |
| 2459 | if (TF_GetCode(status) != TF_OK) { |
| 2460 | TF_DeleteBuffer(ret); |
| 2461 | return nullptr; |
| 2462 | } |
| 2463 | return ret; |
| 2464 | } |
| 2465 | |
| 2466 | TF_Buffer* TF_GetRegisteredKernelsForOp(const char* name, TF_Status* status) { |
| 2467 | tensorflow::KernelList kernel_list = |