From 86f40f352e963be061f4ab722ea0ba99789f96c5 Mon Sep 17 00:00:00 2001 From: Smirnov Egor Date: Thu, 2 Sep 2021 21:28:04 +0300 Subject: [PATCH] add tests for resize+concat optimization --- testdata/dnn/tensorflow/generate_tf_models.py | 8 ++++++++ .../tensorflow/resize_concat_optimization_in.npy | Bin 0 -> 488 bytes .../tensorflow/resize_concat_optimization_net.pb | Bin 0 -> 600 bytes .../tensorflow/resize_concat_optimization_out.npy | Bin 0 -> 848 bytes 4 files changed, 8 insertions(+) create mode 100644 testdata/dnn/tensorflow/resize_concat_optimization_in.npy create mode 100644 testdata/dnn/tensorflow/resize_concat_optimization_net.pb create mode 100644 testdata/dnn/tensorflow/resize_concat_optimization_out.npy diff --git a/testdata/dnn/tensorflow/generate_tf_models.py b/testdata/dnn/tensorflow/generate_tf_models.py index 06b88aaaa..c83778e69 100644 --- a/testdata/dnn/tensorflow/generate_tf_models.py +++ b/testdata/dnn/tensorflow/generate_tf_models.py @@ -1019,6 +1019,14 @@ def pad_depth(x, desired_channels): input_down = tf.image.resize(conv, size=[hi, wi], method=0, name='resize_down') save(inp, input_down, 'resize_bilinear_down') ################################################################################ +inp = tf.placeholder(tf.float32, [1, None, None, 3], 'input') +biased = tf.nn.bias_add(inp, [1, 2, 3], data_format='NHWC') +resized1 = tf.image.resize(biased, [5, 6]) +concat = tf.concat([resized1, biased], 3) +# blob = np.random.standard_normal([1, 5, 6, 3]).astype(tf.float32.as_numpy_dtype()) +# writeBlob(blob, 'resize_concat_optimization_in') +save(inp, concat, 'resize_concat_optimization', optimize=False, is_gen_data=False) +################################################################################ # Uncomment to print the final graph. # with tf.gfile.FastGFile('fused_batch_norm_net.pb', 'rb') as f: diff --git a/testdata/dnn/tensorflow/resize_concat_optimization_in.npy b/testdata/dnn/tensorflow/resize_concat_optimization_in.npy new file mode 100644 index 0000000000000000000000000000000000000000..a56e39e32e38dc6d2557428443b832e2fd9844ca GIT binary patch literal 488 zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+l>qoAIaUsO_*m=~X4l#&V(cT3DEP6dh= zXCxM+0{I$-Its=*3Z^;=W|}$*wF+baF4t!z_P+Bg>=ope?-%brWdGG-(f%tQ|Mp2# zF4*s?nQDJvnydZcx&wAr=Q#Esw9wvnQL}sh9j(Lrtz~)b{SK+vcQa4hC&n$k&+vWW zeii9S_Wn{l_N&ca*e5)myx(#TyS=dxiGy zZ>Q|5ZoIJnQsL%(>T6fqx!pZ(Z&Sl+7vSqu z+FondPW$RD-|VK$+`R9CY0ZB59TN5exkmQ&>s#&5-F|ErYs+MJa-WD@Yx}`{Q@K?4 zCp_2Q_w(`LePKE!cJIoB?5`Uv-FM<%-2Tfg|Mz(`-rF;$Qg`39@}m8R&L7yA2)ZT(66V%JF60{~L)$zA{e literal 0 HcmV?d00001 diff --git a/testdata/dnn/tensorflow/resize_concat_optimization_net.pb b/testdata/dnn/tensorflow/resize_concat_optimization_net.pb new file mode 100644 index 0000000000000000000000000000000000000000..27256c1e8fafc0cb8ff3be2619b0a2e510c30e02 GIT binary patch literal 600 zcmZ{h-%GaQ>had~o<60oj}{V=vC`L7#@`(8m(n%m!*Z+D@h*_@8Sw zwvLS%5^_iGzW2EcJONYbWLgGS*JwW^B9w8KL^9`>U=^{LNIdT~(PReN?02=c9FtLQ zgB6vZI^1jrPRD1fsH@It&$Az4)5|v7kqqFIIYkY63@|yJ!4n(5f(jd(8ROSVm z7lgZju}H?jM9pLpgfcCcFVF=i%+gTkFYC0#Ozj#j<+$ph;n#H!q-Riwp;2Wb(?6m` go;%*Z-E7cTnTwip`XQRG$pLC370vY2Y1I;vKd-x@`Tzg` literal 0 HcmV?d00001 diff --git a/testdata/dnn/tensorflow/resize_concat_optimization_out.npy b/testdata/dnn/tensorflow/resize_concat_optimization_out.npy new file mode 100644 index 0000000000000000000000000000000000000000..759289158833df27595fe5e4aed267da4673f785 GIT binary patch literal 848 zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+l>qoAIaUsO_*m=~X4l#&V(cT3DEP6dh= zXCxM+0{I$-Itpew3Z_6(Q%9jzfegT}R|V+gD+dY`>)3#$nx& zTlONq_S*F*U$$RbD(bNKU9!WGa1Mv7OL!bM=W{!FaV@l$Xbf~n;ns6F>hQ&0{=S}r z=-n8H{%dj$vn4OvOFV6~=g?DgaG9;+aNxR=Lshkm!!>0NhqvCG4)XKP+i$B{YcJCL z%ASe+y!}xpeuq7CRUDp1YCHIOa5$v#D>(dhvU5n&Fm;ehVs_wortVPE@y}l3n~KA( zTa69}LRsy*|D-qw@9B5gcA>*Tadn)-j8}d3(-!vGEBP`w