- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for TF_OpKernelConstruction_GetAttrSize (0.53 sec)
-
tensorflow/c/kernels.h
// (4) If attr_type == TF_ATTR_SHAPE // then total_size is the cumulative number // of dimensions of all shapes in the list. // (5) Otherwise, total_size is undefined. TF_CAPI_EXPORT extern void TF_OpKernelConstruction_GetAttrSize( TF_OpKernelConstruction* ctx, const char* attr_name, int32_t* list_size, int32_t* total_size, TF_Status* status); // Interprets the named kernel construction attribute as a TF_DataType and
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_experimental.h
// fills in `vals` with the size of each dimension. `vals` must point to an // array of length at least `max_values` (ideally set to total_size from // TF_OpKernelConstruction_GetAttrSize(ctx, attr_name, &list_size, // &total_size)). TF_CAPI_EXPORT extern void TF_OpKernelConstruction_GetAttrTensorShape( TF_OpKernelConstruction* ctx, const char* attr_name, int64_t* dims,
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.cc
auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); absl::Status s(tsl::StatusFromTF_Status(status)); cc_ctx->CtxFailure(s); } void TF_OpKernelConstruction_GetAttrSize(TF_OpKernelConstruction* ctx, const char* attr_name, int32_t* list_size, int32_t* total_size,
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
do { \ int32_t list_size, total_size; \ TF_OpKernelConstruction_GetAttrSize(ctx, attr_name, &list_size, \ &total_size, status); \ EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); \
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 50.4K bytes - Viewed (0)