- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for TF_InputRange (0.16 sec)
-
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) -
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) -
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)