- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for TFE_OpGetAttrType (0.11 sec)
-
tensorflow/c/eager/c_api.h
TF_CAPI_EXPORT extern TF_AttrType TFE_OpGetAttrType(TFE_Op* op, const char* attr_name, unsigned char* is_list, TF_Status* status); // Get an attribute type given an op name; a fusion of TFE_NewOp and // TFE_OpGetAttrType for use from Python without the overhead of the individual
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Thu Apr 27 21:07:00 UTC 2023 - 22.8K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
TF_AttrType ret; TFE_Op* op = TFE_NewOp(ctx, op_or_function_name, status); if (status->status.ok()) { ret = TFE_OpGetAttrType(op, attr_name, is_list, status); } else { ret = TF_ATTR_INT; // Same dummy return as TFE_OpGetAttrType. } TFE_DeleteOp(op); return ret; } void TFE_OpSetAttrString(TFE_Op* op, const char* attr_name, const void* value,
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 43.9K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test.cc
TFE_OpAddAttrs(copy_op, attributes); unsigned char is_list = 0; ASSERT_EQ(TF_ATTR_TYPE, TFE_OpGetAttrType(copy_op, "dtype", &is_list, status)); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); ASSERT_EQ(TF_ATTR_SHAPE, TFE_OpGetAttrType(copy_op, "shape", &is_list, status)); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Thu Aug 03 20:50:20 UTC 2023 - 94.6K bytes - Viewed (0)