- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for TF_NumOutputs (0.24 sec)
-
tensorflow/c/kernels.h
// TF_NumInputs returns the number of inputs available in ctx. TF_CAPI_EXPORT extern int TF_NumInputs(TF_OpKernelContext* ctx); // TF_NumOutputs returns the number of outputs to be placed in *ctx by the // kernel. TF_CAPI_EXPORT extern int TF_NumOutputs(TF_OpKernelContext* ctx); // Retrieves the ith input from ctx. If TF_GetCode(status) is TF_OK, *tensor is
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 09 22:46:22 UTC 2024 - 24.6K bytes - Viewed (0) -
tensorflow/c/kernels.cc
} int TF_NumInputs(TF_OpKernelContext* ctx) { auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); return cc_ctx->num_inputs(); } int TF_NumOutputs(TF_OpKernelContext* ctx) { auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); return cc_ctx->num_outputs(); } void TF_GetInput(TF_OpKernelContext* ctx, int i, TF_Tensor** tensor,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 36K bytes - Viewed (0) -
tensorflow/c/kernels_test.cc
// and num_outputs. Various functions on TF_OpKernelContext are also // exercised. auto my_compute_func = [](void* kernel, TF_OpKernelContext* ctx) { num_inputs = TF_NumInputs(ctx); num_outputs = TF_NumOutputs(ctx); TF_Tensor* input = nullptr; TF_Status* s = TF_NewStatus(); TF_GetInput(ctx, 0, &input, s); EXPECT_EQ(TF_OK, TF_GetCode(s)) << "Failed to get input: " << TF_Message(s);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 50.4K bytes - Viewed (0)