Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for TF_NumInputs (0.33 sec)

  1. 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);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 50.4K bytes
    - Viewed (0)
Back to top