DAY33（graph optimization跑inception）

最新推荐文章于 2021-03-11 14:59:37 发布

sinat_39296613

最新推荐文章于 2021-03-11 14:59:37 发布

阅读量118

点赞数

分类专栏：工作日志

本文链接：https://blog.csdn.net/sinat_39296613/article/details/108492300

版权

工作日志专栏收录该内容

32 篇文章 0 订阅

订阅专栏

查看tensorflow版本

pip show tensorflow

修改脚本的可执行权限

chmod +x test.sh

~/.cshrc

alias rm 'rm -i'
alias cp 'cp -i'
alias mv 'mv -i'
alias setprompt 'set prompt="%n@%m ${cwd} >"'
setprompt# to set the initial prompt
alias cd 'chdir \!* && setprompt'
#set prompt="%h:%n@%m:%c >> "
source /arm/tools/setup/init/tcsh
module load core platform/lsf/10.1sp4
module load core arm/cluster/2.0
module load eda synopsys/scl/2017.12
module load swdev python/python/3.6.1
module load swdev python/python/3.6.5
module load swdev jetbrains/pycharm/2018-1.4
module load swdev gnu/gcc/5.5.0
#module load swdev python/anaconda3_py3.6.1/5.2.0 
module load swdev git/git/2.17.1
module load swdev microsoft/vscode/1.42.1
setenv DISPLAY szc-etx-lin1.armchina.com:140.0
setenv PATH /home/ruolu01/.local/bin:$PATH
setenv LD_LIBRARY_PATH /arm/tools/gnu/gcc/5.5.0/rhe7-x86_64/lib64/:$LD_LIBRARY_PATH
setenv PATH /usr/local/cuda/bin:$PATH
setenv PATH /usr/local/cuda/lib64:$PATH
setenv LD_LIBRARY_PATH /usr/local/cuda/lib64:$LD_LIBRARY_PATH

setenv LD_LIBRARY_PATH /arm/tools/python/python/3.6.5/rhe7-x86_64/bin/python:$LD_LIBRARY_PATH

pip show tensorflow

plugin 项目

在这里插入图片描述

1.plugin mechanism design

a. naming convention
b. @register_plugin(type, version)

Plugin base class design

Tensor: numpy.array(),Tensorflow Variable, Torch Tensor
precision
params=int,float,…
weight_shape=[]
weight_type=[]
weight_offset=xx
weight_size=xx
class Base(object):

class Base(object):
	def __init __(self):
		pass
	def forward(self,tensor1,tensor2,...):
 		return out_tensors
 	def quantization_forward(self,tensor1,...):
 		if self.precision==int8:

		else:
		return ...	
	def quantize()
class TFBase(Base):
	pass
class PyTorchBase(Base):
	pass
class Plugin(Base):
	optype=xxx