Lines Matching refs:TF_OpKernelContext
32 void (*compute_function)(void*, TF_OpKernelContext*);
39 void (*compute_func)(void*, TF_OpKernelContext*), in TF_NewKernelBuilder() argument
65 void (*compute_func)(void*, TF_OpKernelContext*), in COpKernel() argument
77 (*compute_func_)(c_kernel_, reinterpret_cast<TF_OpKernelContext*>(ctx)); in Compute()
87 void (*compute_func_)(void*, TF_OpKernelContext* context);
123 int TF_NumInputs(TF_OpKernelContext* ctx) { in TF_NumInputs()
128 int TF_NumOutputs(TF_OpKernelContext* ctx) { in TF_NumOutputs()
133 void TF_GetInput(TF_OpKernelContext* ctx, int i, TF_Tensor** tensor, in TF_GetInput()
147 void TF_SetOutput(TF_OpKernelContext* ctx, int i, const TF_Tensor* tensor, in TF_SetOutput()
170 void TF_OpKernelContext_Failure(TF_OpKernelContext* ctx, TF_Status* status) { in TF_OpKernelContext_Failure()
192 TF_DataType TF_ExpectedOutputDataType(TF_OpKernelContext* ctx, int i) { in DEFINE_TF_GETATTR()
197 int64_t TF_StepId(TF_OpKernelContext* ctx) { in TF_StepId()