feat: Add converter for reflection pad 1d and 2d operation

Signed-off-by: Dheeraj Peri <peri.dheeraj@gmail.com>
pytorch · Feb 24, 2022 · 2484a43 · 2484a43
1 parent 7223fc8
commit 2484a43
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -48,7 +48,7 @@ If you would like to build outside a docker container, please follow the section
 // Set input datatypes. Allowerd options torch::{kFloat, kHalf, kChar, kInt32, kBool}
 // Size of input_dtypes should match number of inputs to the network.
 // If input_dtypes is not set, default precision follows traditional PyT / TRT rules
-auto input = torch_tensorrt::Input(dims, torch::kHalf)
+auto input = torch_tensorrt::Input(dims, torch::kHalf);
 auto compile_settings = torch_tensorrt::ts::CompileSpec({input});
 // FP16 execution
 compile_settings.enabled_precisions = {torch::kHalf};

diff --git a/core/conversion/converters/BUILD b/core/conversion/converters/BUILD
@@ -71,6 +71,7 @@ cc_library(
         "impl/pooling.cpp",
         "impl/quantization.cpp",
         "impl/reduce.cpp",
+        "impl/reflection_pad.cpp",
         "impl/replication_pad.cpp",
         "impl/select.cpp",
         "impl/shuffle.cpp",

diff --git a/tests/core/conversion/converters/BUILD b/tests/core/conversion/converters/BUILD
@@ -83,6 +83,10 @@ converter_test(
     name = "test_reduce",
 )
 
+converter_test(
+    name = "test_reflection_pad",
+)
+
 converter_test(
     name = "test_replication_pad",
 )