Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TF_InputDatatype (0.25 sec)

  1. tensorflow/c/kernels.h

                                             TF_InputRange_Args* args);
    
    // Returns the data type of the index-th input. If index < 0 or index >=
    // TF_NumInputs(ctx), the program aborts.
    TF_CAPI_EXPORT extern TF_DataType TF_InputDatatype(TF_OpKernelContext* ctx,
                                                       int index);
    
    // Sets the ith output of ctx to tensor. If TF_GetCode(status) is anything but
    // TF_OK, ctx is left unmodified.
    //
    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.cc

      auto status = cc_ctx->op_kernel().InputRange(name, &start, &stop);
      args->start = start;
      args->stop = stop;
      tensorflow::Set_TF_Status_from_Status(args->status, status);
    }
    
    TF_DataType TF_InputDatatype(TF_OpKernelContext* ctx, int index) {
      auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx);
      CHECK_GE(index, 0);                     // Crash OK
      CHECK_LT(index, cc_ctx->num_inputs());  // Crash OK
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 22:53:47 UTC 2024
    - 36K bytes
    - Viewed (0)
Back to top