Exemplo n.º 1
0
func init() {
	var l LSTM
	serializer.RegisterTypedDeserializer(l.SerializerType(), DeserializeLSTM)

	var lg lstmGate
	serializer.RegisterTypedDeserializer(lg.SerializerType(), deserializeLSTMGate)
}
Exemplo n.º 2
0
func init() {
	var b BlockSeqFunc
	serializer.RegisterTypedDeserializer(b.SerializerType(), DeserializeBlockSeqFunc)
}
Exemplo n.º 3
0
func init() {
	var b Bidirectional
	serializer.RegisterTypedDeserializer(b.SerializerType(), DeserializeBidirectional)
}
Exemplo n.º 4
0
func init() {
	var n NetworkBlock
	serializer.RegisterTypedDeserializer(n.SerializerType(), DeserializeNetworkBlock)
}
Exemplo n.º 5
0
func init() {
	serializer.RegisterDeserializer(serializerTypeSigmoid,
		func(d []byte) (serializer.Serializer, error) {
			return &Sigmoid{}, nil
		})
	serializer.RegisterDeserializer(serializerTypeReLU,
		func(d []byte) (serializer.Serializer, error) {
			return &ReLU{}, nil
		})
	serializer.RegisterDeserializer(serializerTypeHyperbolicTangent,
		func(d []byte) (serializer.Serializer, error) {
			return &HyperbolicTangent{}, nil
		})
	serializer.RegisterTypedDeserializer(serializerTypeConvLayer,
		DeserializeConvLayer)
	serializer.RegisterTypedDeserializer(serializerTypeDenseLayer,
		DeserializeDenseLayer)
	serializer.RegisterTypedDeserializer(serializerTypeNetwork,
		DeserializeNetwork)
	serializer.RegisterTypedDeserializer(serializerTypeBorderLayer,
		DeserializeBorderLayer)
	serializer.RegisterTypedDeserializer(serializerTypeSoftmaxLayer,
		DeserializeSoftmaxLayer)
	serializer.RegisterTypedDeserializer(serializerTypeLogSoftmaxLayer,
		DeserializeLogSoftmaxLayer)
	serializer.RegisterTypedDeserializer(serializerTypeMaxPoolingLayer,
		DeserializeMaxPoolingLayer)
	serializer.RegisterTypedDeserializer(serializerTypeUnstackLayer,
		DeserializeUnstackLayer)
	serializer.RegisterTypedDeserializer(serializerTypeRescaleLayer,
		DeserializeRescaleLayer)
	serializer.RegisterTypedDeserializer(serializerTypeDropoutLayer,
		DeserializeDropoutLayer)
	serializer.RegisterTypedDeserializer(serializerTypeVecRescaleLayer,
		DeserializeVecRescaleLayer)
	serializer.RegisterTypedDeserializer(serializerTypeGaussNoiseLayer,
		DeserializeGaussNoiseLayer)
}
Exemplo n.º 6
0
func init() {
	var s StateOutBlock
	serializer.RegisterTypedDeserializer(s.SerializerType(), DeserializeStateOutBlock)
}
Exemplo n.º 7
0
func init() {
	var g GRU
	serializer.RegisterTypedDeserializer(g.SerializerType(), DeserializeGRU)
}
Exemplo n.º 8
0
func init() {
	var n NetworkSeqFunc
	serializer.RegisterTypedDeserializer(n.SerializerType(), DeserializeNetworkSeqFunc)
}