From c162a6d671d53dc68156551b9e994bb0fb42af0f Mon Sep 17 00:00:00 2001
From: lianghao <lianghao24@hisilicon.com>
Date: Mon, 18 Oct 2021 20:05:29 +0800
Subject: [PATCH] DT_VARIANT

---
 parser/common/types_map.h | 1 +
 1 file changed, 1 insertion(+)

diff --git a/parser/common/types_map.h b/parser/common/types_map.h
index 082d607..3e75990 100644
--- a/parser/common/types_map.h
+++ b/parser/common/types_map.h
@@ -45,6 +45,7 @@ static map<int32_t, int32_t> GE_TENSORFLOW_DATA_TYPE_MAP = {
     {ge::DataType::DT_QINT32, domi::tensorflow::DT_QINT32},
     {ge::DataType::DT_QUINT8, domi::tensorflow::DT_QUINT8},
     {ge::DataType::DT_QUINT16, domi::tensorflow::DT_QUINT16},
+    {ge::DataType::DT_VARIANT, domi::tensorflow::DT_VARIANT},
     {ge::DataType::DT_DUAL, domi::tensorflow::DT_INVALID},
     {ge::DataType::DT_DUAL_SUB_INT8, domi::tensorflow::DT_INVALID},
     {ge::DataType::DT_DUAL_SUB_UINT8, domi::tensorflow::DT_INVALID},