From 70f2abed9a76d48a1d1a9e8f953d4e3dda06cda7 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <yamanu@xilinx.com>
Date: Wed, 4 Dec 2019 14:32:03 +0000
Subject: [PATCH] [Notebook] add qnt annot example to brevitas-network-import
nb
---
notebooks/brevitas-network-import.ipynb | 55 +++++++++++++++++++++++--
1 file changed, 51 insertions(+), 4 deletions(-)
diff --git a/notebooks/brevitas-network-import.ipynb b/notebooks/brevitas-network-import.ipynb
index 9099cb97f..404242908 100644
--- a/notebooks/brevitas-network-import.ipynb
+++ b/notebooks/brevitas-network-import.ipynb
@@ -506,7 +506,7 @@
},
{
"cell_type": "code",
- "execution_count": 12,
+ "execution_count": 22,
"metadata": {},
"outputs": [
{
@@ -518,7 +518,7 @@
"op_type: \"MatMul\""
]
},
- "execution_count": 12,
+ "execution_count": 22,
"metadata": {},
"output_type": "execute_result"
}
@@ -538,7 +538,7 @@
},
{
"cell_type": "code",
- "execution_count": 13,
+ "execution_count": 23,
"metadata": {},
"outputs": [
{
@@ -553,7 +553,7 @@
" [-1., 1., 1., ..., -1., -1., 1.]], dtype=float32)"
]
},
- "execution_count": 13,
+ "execution_count": 23,
"metadata": {},
"output_type": "execute_result"
}
@@ -562,6 +562,53 @@
"model.get_initializer(model.graph.node[9].input[1])"
]
},
+ {
+ "cell_type": "markdown",
+ "metadata": {},
+ "source": [
+ "We can also examine the quantization annotations and shapes of various tensors using the convenience functions provided by ModelWrapper."
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 24,
+ "metadata": {},
+ "outputs": [
+ {
+ "data": {
+ "text/plain": [
+ "<DataType.BIPOLAR: 8>"
+ ]
+ },
+ "execution_count": 24,
+ "metadata": {},
+ "output_type": "execute_result"
+ }
+ ],
+ "source": [
+ "model.get_tensor_datatype(model.graph.node[9].input[1])"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 25,
+ "metadata": {},
+ "outputs": [
+ {
+ "data": {
+ "text/plain": [
+ "[784, 1024]"
+ ]
+ },
+ "execution_count": 25,
+ "metadata": {},
+ "output_type": "execute_result"
+ }
+ ],
+ "source": [
+ "model.get_tensor_shape(model.graph.node[9].input[1])"
+ ]
+ },
{
"cell_type": "markdown",
"metadata": {},
--
GitLab