- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for TF_OperationGetAttrShape (0.22 sec)
-
tensorflow/c/c_api_test.cc
TF_SetAttrShape(desc, "v", nullptr, -1); auto oper = TF_FinishOperation(desc, s_); ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_); EXPECT_TF_META("v", -1, TF_ATTR_SHAPE, -1); TF_OperationGetAttrShape(oper, "v", nullptr, 10, s_); EXPECT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_); // Partially specified shape const int64_t partial_shape[] = {17, -1};
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 96.9K bytes - Viewed (0) -
tensorflow/c/c_api.h
// `values` must point to an array of length at least `num_dims` (ideally set to // TF_Attr_Meta.size from TF_OperationGetAttrMetadata(oper, attr_name)). TF_CAPI_EXPORT extern void TF_OperationGetAttrShape(TF_Operation* oper, const char* attr_name, int64_t* value,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Oct 26 21:08:15 UTC 2023 - 82.3K bytes - Viewed (0) -
tensorflow/c/c_api.cc
DEFINE_GETATTR(TF_OperationGetAttrBool, unsigned char, bool, b); DEFINE_GETATTR(TF_OperationGetAttrType, TF_DataType, DataType, type); #undef DEFINE_GETATTR void TF_OperationGetAttrShape(TF_Operation* oper, const char* attr_name, int64_t* value, int num_dims, TF_Status* status) { PartialTensorShape shape; status->status =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 102.3K bytes - Viewed (0)