Searched refs:TF_SetOutput (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/c/ |
| D | kernels_test.cc | 172 TF_SetOutput(ctx, 0, input, s); in TEST()
175 TF_SetOutput(ctx, 24, input, s); in TEST()
|
| D | kernels.h | 110 TF_CAPI_EXPORT extern void TF_SetOutput(TF_OpKernelContext* ctx, int i,
|
| D | kernels.cc | 147 void TF_SetOutput(TF_OpKernelContext* ctx, int i, const TF_Tensor* tensor, in TF_SetOutput() function
|
| /external/tensorflow/tensorflow/c/kernels/ |
| D | bitcast_op.cc | 125 TF_SetOutput(ctx, 0, output, status); in BitcastOp_Compute()
|