- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for TF_InputDatatype (0.17 sec)
-
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) -
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)