- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for TF_IsHostMemoryOutput (0.8 sec)
-
tensorflow/c/kernels.h
TF_Status* status); // Returns true if the ith output is allocated in host memory. If i < 0 or i >= // TF_NumOutputs(ctx), the program aborts. TF_CAPI_EXPORT extern bool TF_IsHostMemoryOutput(TF_OpKernelContext* ctx, int i, TF_Status* status); // Returns the step ID of the given context. TF_CAPI_EXPORT extern int64_t TF_StepId(TF_OpKernelContext* ctx);
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
TF_SetStatus(status, TF_OUT_OF_RANGE, "input index out of range"); return false; } TF_SetStatus(status, TF_OK, ""); return cc_ctx->input_memory_type(i) == tensorflow::HOST_MEMORY; } bool TF_IsHostMemoryOutput(TF_OpKernelContext* ctx, int i, TF_Status* status) { auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); if (i < 0 || i >= cc_ctx->num_outputs()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 36K bytes - Viewed (0)