New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[PluggableDevice] Kernel C API enhancement for retrieving attributes #44017
[PluggableDevice] Kernel C API enhancement for retrieving attributes #44017
Conversation
Looks good to me. I added API review label so that API owners can take a look as well |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
(for tf-api-owners)
// | ||
// If the attribute could not be found or could not be interpreted as | ||
// bool, *status is populated with an error. | ||
TF_CAPI_EXPORT extern void TF_OpKernelConstruction_GetAttrString( |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
So should the type of val
be char* val
instead?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
for @tensorflow/api-owners
tensorflow/c/kernels_test.cc
Outdated
|
||
// Helper macros for the TF_OpKernelConstruction_GetAttr* tests. | ||
#define EXPECT_TF_SIZE(attr_name, expected_list_size, expected_total_size) \ | ||
\ |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Remove this line. It just contains \
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
done
…ieving attributes PiperOrigin-RevId: 346844467 Change-Id: I9d2121c3ea2402df51879852d0bade21577b3d9b
…el_c_api PiperOrigin-RevId: 347042144 Change-Id: I05c7587c7a491199f8ba24109eaf4f6d1bc83c72
Adding more attribution C API for pluggable device kernels implementation.