tensor_descriptor.h

Go to the documentation of this file.

#ifndef BLACKCAT_NEURALNETWORKS_POLYMORPHIC_LAYERS_TENSOR_DESCRIPTOR_H_
#define BLACKCAT_NEURALNETWORKS_POLYMORPHIC_LAYERS_TENSOR_DESCRIPTOR_H_

#include "../../../blackcat/tensors.h"

namespace bc {
namespace nn {

template<
        class ValueType,
        class SystemTag,
        class TensorDimension,
        class AllocatorType>
struct tensor_descriptor {

    using value_type = ValueType;
    using system_tag = SystemTag;
    using allocator_type = AllocatorType;

    using tensor_dim = bc::traits::Integer<TensorDimension::value>;
    using batched_tensor_dim = bc::traits::Integer<tensor_dim::value + 1>;

    using shape_type = bc::Dim<tensor_dim::value>;
    using batched_shape_type = bc::Dim<batched_tensor_dim::value>;

    using tensor_type = bc::Tensor<
        tensor_dim::value,
        value_type,
        allocator_type>;

    using batched_tensor_type = bc::Tensor<
        batched_tensor_dim::value,
        value_type,
        allocator_type>;
};

}
}


#endif