Fix documentation of CropMirrorNormalize dtype argument (NVIDIA#3439)

Signed-off-by: Joaquin Anton <janton@nvidia.com>
cyyever · Jan 23, 2022 · 484d373 · 484d373
1 parent d003c52
commit 484d373
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/dali/operators/image/crop/crop_mirror_normalize.cc b/dali/operators/image/crop/crop_mirror_normalize.cc
@@ -42,8 +42,7 @@ Normalization takes the input images and produces the output by using the follow
        R"code(Output data type.
 
 Supported types: ``FLOAT``, ``FLOAT16``, ``INT8``, ``UINT8``.
-
-If not set, the input type is used.)code", DALI_FLOAT)
+)code", DALI_FLOAT)
   .DeprecateArgInFavorOf("output_dtype", "dtype")  // deprecated since 0.24dev
   .AddOptionalArg("output_layout",
     R"code(Tensor data layout for the output.)code", TensorLayout("CHW"))

diff --git a/dali/operators/image/crop/crop_mirror_normalize.h b/dali/operators/image/crop/crop_mirror_normalize.h
@@ -163,8 +163,7 @@ class CropMirrorNormalize : public Operator<Backend> {
   void SetupCommonImpl(const workspace_t<Backend> &ws) {
     const auto &input = ws.template InputRef<Backend>(0);
     input_type_ = input.type();
-    if (output_type_ == DALI_NO_TYPE)
-      output_type_ = input_type_;
+    assert(output_type_ != DALI_NO_TYPE);
 
     auto in_shape = input.shape();
     input_layout_ = input.GetLayout();