Add supported layout info to Crop, CropMirrorNormalize and Flip

Signed-off-by: Joaquin Anton <janton@nvidia.com>
NVIDIA · Mar 7, 2022 · fdc5c39 · fdc5c39
1 parent d2c62c5
commit fdc5c39
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 3 deletions.
diff --git a/dali/operators/generic/flip.cc b/dali/operators/generic/flip.cc
@@ -1,4 +1,4 @@
-// Copyright (c) 2019-2021, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+// Copyright (c) 2019-2022, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 //
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
@@ -30,7 +30,9 @@ and depthwise).)code")
     .AddOptionalArg("horizontal", R"code(Flip the horizontal  dimension.)code", 1, true)
     .AddOptionalArg("vertical", R"code(Flip the vertical dimension.)code", 0, true)
     .AddOptionalArg("depthwise", R"code(Flip the depthwise dimension.)code", 0, true)
-    .InputLayout({"FDHWC", "FHWC", "DHWC", "HWC", "FCDHW", "FCHW", "CDHW", "CHW"})
+    .InputLayout(0, {"HWC", "CHW", "DHWC",
+                     "FHWC", "FCHW", "CDHW", "CFHW",
+                     "FDHWC", "FCDHW", "CFDHW"})
     .AllowSequences()
     .SupportVolumetric();
 

diff --git a/dali/operators/image/crop/crop.cc b/dali/operators/image/crop/crop.cc
@@ -1,4 +1,4 @@
-// Copyright (c) 2019, NVIDIA CORPORATION. All rights reserved.
+// Copyright (c) 2019-2022, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 //
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
@@ -27,6 +27,9 @@ DALI_SCHEMA(Crop)
 (upper left corner).)code")
     .NumInput(1)
     .NumOutput(1)
+    .InputLayout(0, {"HWC", "CHW", "DHWC",
+                     "FHWC", "FCHW", "CDHW", "CFHW",
+                     "FDHWC", "FCDHW", "CFDHW"})
     .AllowSequences()
     .SupportVolumetric()
     .AddOptionalArg<DALIImageType>("image_type", "Image type", nullptr)

diff --git a/dali/operators/image/crop/crop_mirror_normalize.cc b/dali/operators/image/crop/crop_mirror_normalize.cc
@@ -34,6 +34,9 @@ Normalization takes the input images and produces the output by using the follow
 )code")
   .NumInput(1)
   .NumOutput(1)
+  .InputLayout(0, {"HWC", "CHW", "DHWC",
+                   "FHWC", "FCHW", "CDHW", "CFHW",
+                   "FDHWC", "FCDHW", "CFDHW"})
   .AllowSequences()
   .SupportVolumetric()
   .AddOptionalArg<DALIImageType>("image_type", "Image type", nullptr)