- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for TF_OpKernelConstruction_GetName (0.48 sec)
-
tensorflow/c/kernels/histogram_summary_op.cc
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.h
TF_OpKernelConstruction* ctx, const char* attr_name, TF_Status* status); // Returns the unique operation name for this OpKernel. TF_CAPI_EXPORT extern TF_StringView TF_OpKernelConstruction_GetName( TF_OpKernelConstruction* ctx); // Allocates Tensor for output at given index. Caller takes ownership of // returned TF_Tensor and should deallocate it using TF_DeleteTensor(tensor). //
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* cc_ctx = reinterpret_cast<::tensorflow::OpKernelConstruction*>(ctx); return cc_ctx->HasAttr(attr_name); } TF_StringView TF_OpKernelConstruction_GetName(TF_OpKernelConstruction* ctx) { auto* cc_ctx = reinterpret_cast<tensorflow::OpKernelConstruction*>(ctx); TF_StringView string_view_of_name; string_view_of_name.data = cc_ctx->def().name().data();
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_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); EXPECT_EQ(node_name, candidate_node_name); return s; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 50.4K bytes - Viewed (0)