| 169 | } |
| 170 | |
| 171 | int TF_NumOutputs(TF_OpKernelContext* ctx) { |
| 172 | auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); |
| 173 | return cc_ctx->num_outputs(); |
| 174 | } |
| 175 | |
| 176 | void TF_GetInput(TF_OpKernelContext* ctx, int i, TF_Tensor** tensor, |
| 177 | TF_Status* status) { |