Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for TF_NumInputs (0.13 sec)

  1. tensorflow/c/kernels.h

    // to change at any time.
    TF_CAPI_EXPORT extern SP_Stream TF_GetStream(TF_OpKernelContext* ctx,
                                                 TF_Status* status);
    
    // 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.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jan 09 22:46:22 UTC 2024
    - 24.6K bytes
    - Viewed (0)
  2. tensorflow/c/kernels/merge_summary_op.cc

    void MergeSummaryOp_Compute(void* kernel, TF_OpKernelContext* ctx) {
      tensorflow::Summary s;
      std::unordered_set<tensorflow::string> tags;
      Safe_TF_StatusPtr status(TF_NewStatus());
      for (int input_num = 0; input_num < TF_NumInputs(ctx); ++input_num) {
        TF_Tensor* input;
        TF_GetInput(ctx, input_num, &input, status.get());
        Safe_TF_TensorPtr safe_input_ptr(input);
        if (TF_GetCode(status.get()) != TF_OK) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Mar 31 03:28:11 UTC 2021
    - 4.7K bytes
    - Viewed (0)
  3. tensorflow/c/kernels.cc

            cc_ctx->op_device_context()->stream());
        return c_stream->Handle();
      }
    #endif  // defined(IS_MOBILE_PLATFORM) || defined(IS_SLIM_BUILD)
    }
    
    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) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 22:53:47 UTC 2024
    - 36K bytes
    - Viewed (0)
  4. 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