[PluggableDevice] Add TF_OpKernelConstruction_GetAttrTensor and TF_OpKernelConstruction_GetAttrTensorList #50140
Conversation
…KernelConstruction_GetAttrTensorList
google-ml-butler
bot
added
kokoro:force-run
tensorflow/c/kernels_test.cc
Outdated
| @@ -291,6 +293,133 @@ TEST_F(TestKernelAttr, StringList) { | |||
| CreateAndCallKernelWithAttr(my_create_func, "TestKernelAttrStringList", v); | |||
| } | |||
|
|
|||
| TEST_F(TestKernelAttr, Tensor) { | |||
| struct TensorProtoData : public ::tensorflow::TensorProto { | |||
There was a problem hiding this comment.
This seems to break internal builds since ::tensorflow::TensorProto is marked final (not sure why the OSS tests pass). Could we avoid subclassing TensorProto here?
|
We found a Contributor License Agreement for you (the sender of this pull request), but were unable to find agreements for all the commit author(s) or Co-authors. If you authored these, maybe you used a different email address in the git commits than was used to sign the CLA (login here to double check)? If these were authored by someone else, then they will need to sign a CLA as well, and confirm that they're okay with these being contributed to Google. ℹ️ Googlers: Go here for more info. |
PatriceVignola
force-pushed
the
add-get-attr-tensor
branch
from
09ffd4d
to
83d1e66
Compare
|
@googlebot I fixed it. |
google-ml-butler
bot
added
kokoro:force-run
google-ml-butler
bot
removed
ready to pull
Add TF_OpKernelConstruction_GetAttrTensor and TF_OpKernelConstruction_GetAttrTensorList to the Pluggable Device C API.