def add_matmul(self, input_a_name, input_b_name, output_name, transpose_a=False, transpose_b=False):
layer = caffe_net.LayerParameter(name=output_name, type='MatMul',
bottom=[input_a_name, input_b_name], top=[output_name])
layer.matmul_param(transpose_a, transpose_b)
self.caffe_model.add_layer(layer)
self.data_dict[output_name] = Operators.matmul(self.data_dict[input_a_name], transpose_a,
self.data_dict[input_b_name], transpose_b, output_name)
return output_name
def add_matmul(self, input_a_name, input_b_name, output_name):
layer = caffe_net.LayerParameter(name=output_name, type='MatMul',
bottom=[input_a_name, input_b_name], top=[output_name])
self.caffe_model.add_layer(layer)
self.data_dict[output_name] = Operators.matmul(self.data_dict[input_a_name], self.data_dict[input_b_name], output_name)