Remove the optional parameters from the Pooling layer.

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
This commit is contained in:
Davide Sapienza
2020-04-09 19:20:27 +02:00
parent a8fd4cabf3
commit 50ada3db60
15 changed files with 65 additions and 65 deletions
+3 -3
View File
@@ -32,13 +32,13 @@ function print_output {
out_file=results.log
rm $out_file
# modes=( 1 ) # only FP32
modes=( 1 2 ) # FP32 and FP16
modes=( 1 ) # only FP32
# modes=( 1 2 ) # FP32 and FP16
# modes=( 1 2 3 ) # FP32, FP16 and INT8
for i in "${modes[@]}"
do
# rm *rt
rm *rt
if [ $i -eq 1 ]
then
export TKDNN_MODE=FP32