Exemplo n.º 1
0
 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