| 164 | } |
| 165 | |
| 166 | int TF_NumInputs(TF_OpKernelContext* ctx) { |
| 167 | auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); |
| 168 | return cc_ctx->num_inputs(); |
| 169 | } |
| 170 | |
| 171 | int TF_NumOutputs(TF_OpKernelContext* ctx) { |
| 172 | auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); |