- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for TF_StringView (0.24 sec)
-
tensorflow/c/kernels.cc
return res; } #endif TF_StringView TF_GetOpKernelName(TF_OpKernelContext* ctx) { auto cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); TF_StringView opkernel_name_sv; opkernel_name_sv.data = cc_ctx->op_kernel().name().data(); opkernel_name_sv.len = cc_ctx->op_kernel().name().length(); return opkernel_name_sv; }
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_experimental.h
// 1. Cannot allocate a new temporary variable // 2. Calling plugin allocator failed TF_CAPI_EXPORT extern void TF_TemporaryVariable( TF_OpKernelContext* ctx, TF_DataType dtype, const int64_t* dims, int num_dims, TF_StringView* var_name, void (*plugin_allocator)(TF_OpKernelContext*, TF_Tensor*, TF_DataType, const int64_t*, int, TF_Status*), TF_Status* tf_status);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Aug 07 14:44:39 UTC 2023 - 9.4K bytes - Viewed (0) -
tensorflow/c/kernels.h
// has the same lifetime as the OpKernel. TF_CAPI_EXPORT extern TF_StringView TF_GetOpKernelName(TF_OpKernelContext* ctx); // Returns the default container of the resource manager in OpKernelContext. // // The returned TF_StringView's underlying string is owned by the OpKernel and // has the same lifetime as the OpKernel. TF_CAPI_EXPORT extern TF_StringView TF_GetResourceMgrDefaultContainerName( 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/histogram_summary_op.cc
// kernel computation. struct HistogramSummaryOp { std::string op_node_name; }; void* HistogramSummaryOp_Create(TF_OpKernelConstruction* ctx) { HistogramSummaryOp* kernel = new HistogramSummaryOp; TF_StringView string_view_name = TF_OpKernelConstruction_GetName(ctx); kernel->op_node_name = std::string(string_view_name.data, string_view_name.len); return kernel; } void HistogramSummaryOp_Delete(void* kernel) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 6.5K bytes - Viewed (0) -
tensorflow/c/kernels_experimental.cc
} return var_name; } void TF_TemporaryVariable(TF_OpKernelContext* ctx, TF_DataType dtype, const int64_t* dims, int num_dims, TF_StringView* var_name, void (*allocFunc)(TF_OpKernelContext*, TF_Tensor*, TF_DataType, const int64_t*, int, TF_Status*),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 23 06:12:29 UTC 2024 - 30.9K bytes - Viewed (0) -
tensorflow/c/kernels_test.cc
TF_OpKernelConstruction_GetAttrType(ctx, "SomeDataTypeAttr", &type, status); EXPECT_EQ(TF_OK, TF_GetCode(status)); EXPECT_EQ(TF_FLOAT, type); TF_DeleteStatus(status); // Exercise kernel NodeDef name read TF_StringView name_string_view = TF_OpKernelConstruction_GetName(ctx); std::string node_name = "SomeNodeName"; std::string candidate_node_name = std::string(name_string_view.data, name_string_view.len);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 50.4K bytes - Viewed (0) -
tensorflow/c/c_api.h
// Used to return strings across the C API. The caller does not take ownership // of the underlying data pointer and is not responsible for freeing it. typedef struct TF_StringView { const char* data; size_t len; } TF_StringView; // -------------------------------------------------------------------------- // TF_SessionOptions holds options that can be passed during session creation.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Oct 26 21:08:15 UTC 2023 - 82.3K bytes - Viewed (0)