Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for TF_InputRange (0.16 sec)

  1. tensorflow/c/kernels.h

    // Retrieves the start and stop indices, given the input name. Equivalent to
    // OpKernel::InputRange(). `args` will contain the result indices and status.
    TF_CAPI_EXPORT extern void TF_InputRange(TF_OpKernelContext* ctx,
                                             const char* name,
                                             TF_InputRange_Args* args);
    
    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

        return;
      }
      TF_Tensor* result =
          ::tensorflow::TF_TensorFromTensor(cc_tensor, &status->status);
      if (TF_GetCode(status) == TF_OK) {
        *tensor = result;
      }
    }
    
    void TF_InputRange(TF_OpKernelContext* ctx, const char* name,
                       TF_InputRange_Args* args) {
      auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx);
      int start = -1, stop = -1;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 22:53:47 UTC 2024
    - 36K bytes
    - Viewed (0)
  3. tensorflow/c/kernels_test.cc

        EXPECT_STREQ(sv.data, "input1");
    
        sv = TF_GetOpKernelRequestedInput(ctx, 1);
        EXPECT_STREQ(sv.data, "input2");
    
        TF_InputRange_Args args;
        args.status = s;
        TF_InputRange(ctx, "input3", &args);
        EXPECT_EQ(TF_OK, TF_GetCode(s));
        EXPECT_EQ(args.start, 2);
        EXPECT_EQ(args.stop, 5);
    
        TF_Tensor* output = TF_AllocateTemp(
    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