Skip to content
This repository was archived by the owner on Feb 3, 2025. It is now read-only.

TF-TRT C++ conversion example - #271

Merged
DEKHTIARJonathan merged 2 commits into
tensorflow:masterfrom
tfeher:cpp_conversion_example
Jun 28, 2022
Merged

TF-TRT C++ conversion example#271
DEKHTIARJonathan merged 2 commits into
tensorflow:masterfrom
tfeher:cpp_conversion_example

Conversation

@tfeher

@tfehertfeher commented Sep 22, 2021

Copy link
Copy Markdown
Contributor

C++ conversion example to illustrate conversion using dynamic shape mode.

This PR depends on tensorflow/tensorflow#52012

@tfeher
tfeherforce-pushed the cpp_conversion_example branch from 5cbb42d to c50c2bdCompareSeptember 26, 2021 23:46
@tfeher
tfeher marked this pull request as ready for review September 26, 2021 23:47
@christopherbate

Copy link
Copy Markdown

I think we're trying to follow convention of labeling PR titles with the tag [TF:TRT]

@christopherbatechristopherbate left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Mainly build system / scripts and instructions need to be updated for final version, other things are minor.

Comment threadtftrt/examples/cpp/image-classification/CMakeLists.txt
Comment threadtftrt/examples/cpp/image-classification/README.md Outdated
Comment threadtftrt/examples/cpp/image-classification/README.md Outdated
Comment threadtftrt/examples/cpp/image-classification/README.md Outdated
cd tensorrt/tftrt/examples/cpp/image-classification
mkdir build
cd build
cmake ..

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Update to include -D... when CMakeLists is finalized

Copy link
Copy Markdown
ContributorAuthor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

What is the purpose of the these, do we want to add hints for TF_SOURCE_DIR / TF_INSTALL_DIR?

Comment threadtftrt/examples/cpp/image-classification/main.cc Outdated
Comment threadtftrt/examples/cpp/image-classification/main.cc Outdated
Comment threadtftrt/examples/cpp/image-classification/main.cc Outdated
Comment threadtftrt/examples/cpp/image-classification/main.cc Outdated
Comment threadtftrt/examples/cpp/image-classification/main.cc
@DEKHTIARJonathan
DEKHTIARJonathanforce-pushed the master branch 5 times, most recently from 94c43ee to 114ebe1CompareOctober 20, 2021 00:33
@tfeher
tfeherforce-pushed the cpp_conversion_example branch from c50c2bd to 08a055cCompareOctober 20, 2021 14:51

@tfehertfeher left a comment

Copy link
Copy Markdown
ContributorAuthor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks @christopherbate for the review, I have addressed the issues.

Comment threadtftrt/examples/cpp/image-classification/README.md Outdated
Comment threadtftrt/examples/cpp/image-classification/README.md Outdated
Comment threadtftrt/examples/cpp/image-classification/README.md Outdated
Comment threadtftrt/examples/cpp/image-classification/main.cc Outdated
Comment threadtftrt/examples/cpp/image-classification/main.cc
Comment threadtftrt/examples/cpp/image-classification/mnist.cc Outdated
Comment threadtftrt/examples/cpp/image-classification/mnist.h Outdated
Comment threadtftrt/examples/cpp/image-classification/mnist.h Outdated
Comment threadtftrt/examples/cpp/image-classification/mnist.cc
Comment threadtftrt/examples/cpp/image-classification/main.cc Outdated
@tfeher

Copy link
Copy Markdown
ContributorAuthor

To compile the example, one needs PR tensorflow/tensorflow#52595

@DEKHTIARJonathan
DEKHTIARJonathanforce-pushed the master branch 2 times, most recently from 27ef7b8 to 535c132CompareMay 10, 2022 21:42
@DEKHTIARJonathan

Copy link
Copy Markdown
Contributor

@meena-at-work: for review

@meena-at-workmeena-at-work left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM!

Sign up for freeto subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants

@tfeher@christopherbate@DEKHTIARJonathan@meena-at-work