Skip to content
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

Taught IRGen to support tensor-typed attributes with a scalar string as the value #21492

Merged
merged 1 commit into from Dec 23, 2018

Conversation

mhong
Copy link
Contributor

@mhong mhong commented Dec 21, 2018

string-tensor-typed attributes are usually to represent metadata like file names in a dataset TF op, so scalar tensor support should be sufficient.

Extending the code to also support string-tensors of arbitrary shapes would involve extending createArrayAndSize() and the runtime function swift_tfc_CreateScalarStringTensor(), which adds complexity without sufficient justification for now.

Resolves SR-9555.

@mhong
Copy link
Contributor Author

mhong commented Dec 21, 2018

@swift-ci please test tensorflow

@mhong
Copy link
Contributor Author

mhong commented Dec 21, 2018

Also cc @bgogul and @rxwei as FYI.

@mhong mhong merged commit 77e3dac into apple:tensorflow Dec 23, 2018
@mhong mhong deleted the string_tensor_attr branch December 23, 2018 14:48
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants