Skip to content

[TFTRT - Dynamic Shape Phase 3] Add Dynamic Shape Testing for ConvertResize & Various TF2TRT Node Convert Unittest Improvements - #46376

Merged
copybara-service[bot] merged 1 commit into
tensorflow:masterfrom
DEKHTIARJonathan:jdekhtiar/dynamic_shape/ConvertResize
Jan 25, 2021
Merged

copybara-service[bot] merged 1 commit into
tensorflow:masterfrom
DEKHTIARJonathan:jdekhtiar/dynamic_shape/ConvertResize

Conversation

@DEKHTIARJonathan

@DEKHTIARJonathan DEKHTIARJonathan commented Jan 12, 2021 •

Copy link
Copy Markdown
Contributor

@bixia1 @tfeher for review

ConvertResize now requires a static shape for the tensor representing the resized size to workaround the fact that we can't support shape tensors as inputs yet.
Add dynamic shape test cases for ConvertResize.

Feature Tracker: #45481

@google-ml-butler google-ml-butler Bot added the size:M CL Change Size: Medium label Jan 12, 2021
@google-ml-butler
google-ml-butler Bot requested a review from joker-eph January 12, 2021 20:35
@google-cla google-cla Bot added the cla: yes label Jan 12, 2021
@bixia1
bixia1 self-requested a review January 12, 2021 20:59
@DEKHTIARJonathan

Copy link
Copy Markdown
Contributor Author
INFO: Elapsed time: 198.196s, Critical Path: 190.09s
INFO: 78 processes: 78 local.
INFO: Build completed successfully, 80 total actions
//tensorflow/compiler/tf2tensorrt:convert_graph_test                     PASSED in 14.9s
//tensorflow/compiler/tf2tensorrt:convert_graph_test_gpu                 PASSED in 14.4s
//tensorflow/compiler/tf2tensorrt:convert_nodes_test                     PASSED in 82.8s
//tensorflow/compiler/tf2tensorrt:convert_nodes_test_gpu                 PASSED in 82.1s
//tensorflow/compiler/tf2tensorrt:segment_test                           PASSED in 0.2s
//tensorflow/compiler/tf2tensorrt:segment_test_gpu                       PASSED in 0.1s
//tensorflow/compiler/tf2tensorrt:tensorrt_test_cc                       PASSED in 12.7s
//tensorflow/compiler/tf2tensorrt:tensorrt_test_cc_gpu                   PASSED in 12.8s
//tensorflow/compiler/tf2tensorrt:trt_allocator_test                     PASSED in 0.1s
//tensorflow/compiler/tf2tensorrt:trt_engine_op_test                     PASSED in 12.4s
//tensorflow/compiler/tf2tensorrt:trt_engine_op_test_gpu                 PASSED in 12.4s
//tensorflow/compiler/tf2tensorrt:trt_engine_resource_ops_test           PASSED in 12.2s
//tensorflow/compiler/tf2tensorrt:trt_engine_resource_ops_test_gpu       PASSED in 12.1s
//tensorflow/compiler/tf2tensorrt:trt_lru_cache_test                     PASSED in 0.1s
//tensorflow/compiler/tf2tensorrt:trt_shape_optimization_profiles_test   PASSED in 12.8s
//tensorflow/compiler/tf2tensorrt:trt_shape_optimization_profiles_test_gpu PASSED in 12.5s

Executed 16 out of 16 tests: 16 tests pass.
INFO: Build completed successfully, 80 total actions
****

@DEKHTIARJonathan

Copy link
Copy Markdown
Contributor Author

@bixia1 there seems to be an issue with DynamicShape & IResizeLayer. For now the unittest basically does not test this case, we are investigating. Though this PR can be merged as-is and we will later fix the test when it's fully working

@gbaned gbaned self-assigned this Jan 13, 2021
@gbaned gbaned added the comp:gpu:tensorrt Issues specific to TensorRT label Jan 13, 2021
@DEKHTIARJonathan

Copy link
Copy Markdown
Contributor Author

@bixia1 : we are good to go. All the modifications needed requested are merged and the PR has been squashed.
Please review ;)

@DEKHTIARJonathan DEKHTIARJonathan changed the title [TFTRT - Dynamic Shape Phase 3] Add Dynamic Shape Testing for ConvertResize [TFTRT - Dynamic Shape Phase 3] Add Dynamic Shape Testing for ConvertResize & Various TF2TRT Node Convert Unittest Improvements Jan 21, 2021
@DEKHTIARJonathan

Copy link
Copy Markdown
Contributor Author

@bixia1 I have applied the changes you requested.

I have also updated PR Title and PR Description. If you would prefer smthg else, please give me the exact content you want me to copy/paste. (Or you can also directly edit if allowed by Github).

@bixia1 bixia1 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.

There are two import things in this PR: change to the converter and new test cases. Can we change the PR description to use this:
ConvertResize now requires a static shape for the tensor representing the resized size to workaround the fact that we can't support shape tensors as inputs yet.
Add dynamic shape test cases for ConvertResize.

@DEKHTIARJonathan

Copy link
Copy Markdown
Contributor Author

@bixia1 I updated as you asked

@google-ml-butler google-ml-butler Bot added kokoro:force-run Tests on submitted change ready to pull PR ready for merge process labels Jan 22, 2021
@kokoro-team kokoro-team removed the kokoro:force-run Tests on submitted change label Jan 22, 2021
@copybara-service
copybara-service Bot merged commit 53af6cb into tensorflow:master Jan 25, 2021
@DEKHTIARJonathan
DEKHTIARJonathan deleted the jdekhtiar/dynamic_shape/ConvertResize branch January 28, 2021 22:07
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

cla: yes comp:gpu:tensorrt Issues specific to TensorRT ready to pull PR ready for merge process size:M CL Change Size: Medium

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants