diff --git a/dali/operators/generic/flip.cc b/dali/operators/generic/flip.cc index 4400495a6ab..29ce930639b 100644 --- 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 index e15a7904436..038efc303d6 100644 --- 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("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 index fa73f956262..6ddf2fea252 100644 --- 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("image_type", "Image type", nullptr)