diff options
author | 2018-07-19 19:14:03 -0700 | |
---|---|---|
committer | 2018-07-19 19:14:08 -0700 | |
commit | 3a56de02398b61226924955c6ee4297a4ecb5d45 (patch) | |
tree | aff45cd43e2d4ac57b9d77104b368ae86851dabb /tensorflow/contrib/tensorrt/convert | |
parent | 6155633318d648ce6ad9567ad5163bd9fc763454 (diff) | |
parent | c519794c7cca51d2c75aa53b56a1448804f68647 (diff) |
Merge pull request #20969 from aaroey:fix_plugin_test
PiperOrigin-RevId: 205340359
Diffstat (limited to 'tensorflow/contrib/tensorrt/convert')
-rw-r--r-- | tensorflow/contrib/tensorrt/convert/convert_nodes.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/contrib/tensorrt/convert/convert_nodes.cc b/tensorflow/contrib/tensorrt/convert/convert_nodes.cc index 65fef27533..49e825151a 100644 --- a/tensorflow/contrib/tensorrt/convert/convert_nodes.cc +++ b/tensorflow/contrib/tensorrt/convert/convert_nodes.cc @@ -2588,6 +2588,8 @@ void Converter::register_op_converters() { op_registry_["BatchMatMul"] = ConvertBatchMatMul; op_registry_["TopKV2"] = ConvertTopK; #endif + + plugin_converter_ = ConvertPlugin; } } // namespace |