inputlayer.h

Go to the documentation of this file.

#include <vector>
#include "layer/layer.h"
#include "tensor/tensor.h"
#include "compute/operation.h"

namespace magmadnn {
namespace layer {

template <typename T>
class InputLayer : public Layer<T> {
public:
    InputLayer(op::Operation<T> *input);

    virtual std::vector<op::Operation<T> *> get_weights();

protected:
    void init();

};

template <typename T>
InputLayer<T>* input(op::Operation<T> *input);

}   // layer
}   // magmadnn