Comment added

Change-Id: I3e2d29ef201283de337168d0b82679b63ca2fcf4
apache · Mar 16, 2021 · 65ca107 · 65ca107
1 parent e619c70
commit 65ca107
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/python/tvm/relay/frontend/tflite.py b/python/tvm/relay/frontend/tflite.py
@@ -2335,6 +2335,11 @@ def convert_cast(self, op):
         assert len(input_tensors) == 1, "input tensors length should be 1"
         input_tensor = input_tensors[0]
         in_expr = self.get_expr(input_tensor.tensor_idx)
+
+        # MLIR-based converter outputs no BuiltinOptions for Cast operator. In this
+        # case the output type can be derived from the Cast operator output tensor.
+        # When TOCO converter is used there will be "normal" BuiltinOptions.CastOptions
+        # with output type.
         if op.BuiltinOptions() is not None:
             assert op.BuiltinOptionsType() == BuiltinOptions.CastOptions
             op_options = op.BuiltinOptions()