Home
last modified time | relevance | path

Searched refs:TF_OpKernelContext (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/c/
Dkernels.h40 typedef struct TF_OpKernelContext TF_OpKernelContext; typedef
73 void (*compute_func)(void*, TF_OpKernelContext*),
92 TF_CAPI_EXPORT extern int TF_NumInputs(TF_OpKernelContext* ctx);
96 TF_CAPI_EXPORT extern int TF_NumOutputs(TF_OpKernelContext* ctx);
103 TF_CAPI_EXPORT extern void TF_GetInput(TF_OpKernelContext* ctx, int i,
110 TF_CAPI_EXPORT extern void TF_SetOutput(TF_OpKernelContext* ctx, int i,
120 TF_CAPI_EXPORT extern void TF_OpKernelContext_Failure(TF_OpKernelContext* ctx,
126 TF_OpKernelContext* ctx, int i);
129 TF_CAPI_EXPORT extern int64_t TF_StepId(TF_OpKernelContext* ctx);
Dkernels.cc32 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()
[all …]
Dkernels_test.cc51 static void MyComputeFunc(void* kernel, TF_OpKernelContext* ctx) { in MyComputeFunc()
157 auto my_compute_func = [](void* kernel, TF_OpKernelContext* ctx) { in TEST()
Dc_test.c40 void compute(void* kernel, TF_OpKernelContext* ctx) { in compute()
/external/tensorflow/tensorflow/c/kernels/
Dbitcast_op.cc89 static void BitcastOp_Compute(void* kernel, TF_OpKernelContext* ctx) { in BitcastOp_Compute()