layer_descriptor.h

Go to the documentation of this file.

#ifndef BLACKCAT_NEURALNETWORKS_POLYMORHIC_LAYERS_LAYER_DESCRIPTOR_H_
#define BLACKCAT_NEURALNETWORKS_POLYMORHIC_LAYERS_LAYER_DESCRIPTOR_H_

namespace bc {
namespace nn {

template<class OutputTensorDescriptor>
class Layer_Input_Base;

template<class OutputTensorDescriptor>
struct Layer_Output_Base
{
    using output_value_type = typename OutputTensorDescriptor::value_type;
    using output_system_tag = typename OutputTensorDescriptor::system_tag;
    using output_allocator_type = typename OutputTensorDescriptor::allocator_type;
    using output_tensor_dim = typename OutputTensorDescriptor::tensor_dim;
    using output_shape_type = typename OutputTensorDescriptor::shape_type; 
        
    using output_tensor_type = typename OutputTensorDescriptor::tensor_type;
    using batched_output_tensor_type = typename OutputTensorDescriptor::batched_type;
    using next_layer_type = Layer_Input_Base<OutputTensorDescriptor>;

private:
    using self_type = Layer_Output_Base<OutputTensorDescriptor>;

protected:
    next_layer_type* m_next_layer = nullptr;
    bc::Dim<output_tensor_dim::value> m_output_shape;

public:
    void set_next(next_layer_type& next) { m_next_layer = &next; }
    output_shape_type output_shape() const { return m_output_shape; }
    const next_layer_type& next_layer() const { return *m_next_layer; }
    next_layer_type& next_layer() { return *m_next_layer; }
    virtual ~Layer_Output_Base() {}
};

template<class InputTensorDescriptor>
class Layer_Input_Base
{
    using input_value_type = typename InputTensorDescriptor::value_type;
    using input_system_tag = typename InputTensorDescriptor::system_tag;
    using input_allocator_type = typename InputTensorDescriptor::allocator_type;
    using input_tensor_dim = typename InputTensorDescriptor::tensor_dim;
    using input_shape_type = bc::Dim<input_tensor_dim::value>;
    using input_tensor_type = typename InputTensorDescriptor::type;
    using batched_input_tensor_type = typename InputTensorDescriptor::batched_type;
    using prev_layer_type = Layer_Output_Base<InputTensorDescriptor>;
private:
    using self_type = Layer_Input_Base<InputTensorDescriptor>;

protected:
    prev_layer_type* m_prev_layer;
    input_shape_type m_input_shape;

public:
    void set_prev(prev_layer_type& prev) { m_prev_layer = &prev; }
    input_shape_type input_shape() const { return m_input_shape; }
    const prev_layer_type& prev_layer() const { return *m_prev_layer; }
    prev_layer_type& prev_layer() { return *m_prev_layer; }
    virtual ~Layer_Input_Base() {}
};
        
}
}

#endif