コード例 #1
0
ファイル: tensorflow2caffe.py プロジェクト: yyqgood/bolt
 def add_gelu(self, input_name, output_name):
     layer = caffe_net.LayerParameter(name=output_name, type='Gelu',
                 bottom=[input_name],
                 top=[output_name])
     self.caffe_model.add_layer(layer)
     self.data_dict[output_name] = Operators.gelu(self.data_dict[input_name], output_name)
     return output_name