| 39 | }; |
| 40 | |
| 41 | TF_KernelBuilder* TF_NewKernelBuilder( |
| 42 | const char* op_name, const char* device_name, |
| 43 | void* (*create_func)(TF_OpKernelConstruction*), |
| 44 | void (*compute_func)(void*, TF_OpKernelContext*), |
| 45 | void (*delete_func)(void*)) { |
| 46 | TF_KernelBuilder* result = new TF_KernelBuilder; |
| 47 | result->cc_builder = new ::tensorflow::KernelDefBuilder(op_name); |
| 48 | result->cc_builder->Device(device_name); |
| 49 | result->create_function = create_func; |
| 50 | result->compute_function = compute_func; |
| 51 | result->delete_function = delete_func; |
| 52 | return result; |
| 53 | } |
| 54 | |
| 55 | void TF_DeleteKernelBuilder(TF_KernelBuilder* builder) { |
| 56 | if (builder != nullptr) { |