Searched refs:TF_SetOutput (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/c/ |
| D | kernels_test.cc | 534 TF_SetOutput(ctx, 0, input, s); in TEST()
537 TF_SetOutput(ctx, 24, input, s); in TEST()
833 TF_SetOutput(ctx, 0, output, s); in TEST_F()
865 TF_SetOutput(ctx, 0, output, s); in TEST_F()
902 TF_SetOutput(ctx, 0, output, s); in TEST_F()
|
| D | kernels.h | 166 TF_CAPI_EXPORT extern void TF_SetOutput(TF_OpKernelContext* ctx, int i,
|
| D | kernels.cc | 248 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 | 127 TF_SetOutput(ctx, 0, output, status); in BitcastOp_Compute()
|