BlackCat_Tensors
A GPU-supported autograd and linear algebra library, designed for neural network construction
|
#include <layer_base.h>
Public Types | |
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 = bc::Dim< output_tensor_dim::value > |
using | output_tensor_type = typename OutputTensorDescriptor::type |
using | batched_output_tensor_type = typename OutputTensorDescriptor::batched_type |
using | next_layer_type = Layer_Input_Base< OutputTensorDescriptor > |
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 > |
Public Member Functions | |
void | set_next (next_layer_type &next) |
output_shape_type | output_shape () const |
const next_layer_type & | next_layer () const |
next_layer_type & | next_layer () |
virtual | ~Layer_Output_Base () |
void | set_next (next_layer_type &next) |
output_shape_type | output_shape () const |
const next_layer_type & | next_layer () const |
next_layer_type & | next_layer () |
virtual | ~Layer_Output_Base () |
Protected Attributes | |
next_layer_type * | m_next_layer = nullptr |
bc::Dim< output_tensor_dim::value > | m_output_shape |
using bc::nn::Layer_Output_Base< OutputTensorDescriptor >::batched_output_tensor_type = typename OutputTensorDescriptor::batched_type |
using bc::nn::Layer_Output_Base< OutputTensorDescriptor >::batched_output_tensor_type = typename OutputTensorDescriptor::batched_type |
using bc::nn::Layer_Output_Base< OutputTensorDescriptor >::next_layer_type = Layer_Input_Base<OutputTensorDescriptor> |
using bc::nn::Layer_Output_Base< OutputTensorDescriptor >::next_layer_type = Layer_Input_Base<OutputTensorDescriptor> |
using bc::nn::Layer_Output_Base< OutputTensorDescriptor >::output_allocator_type = typename OutputTensorDescriptor::allocator_type |
using bc::nn::Layer_Output_Base< OutputTensorDescriptor >::output_allocator_type = typename OutputTensorDescriptor::allocator_type |
using bc::nn::Layer_Output_Base< OutputTensorDescriptor >::output_shape_type = typename OutputTensorDescriptor::shape_type |
using bc::nn::Layer_Output_Base< OutputTensorDescriptor >::output_shape_type = bc::Dim<output_tensor_dim::value> |
using bc::nn::Layer_Output_Base< OutputTensorDescriptor >::output_system_tag = typename OutputTensorDescriptor::system_tag |
using bc::nn::Layer_Output_Base< OutputTensorDescriptor >::output_system_tag = typename OutputTensorDescriptor::system_tag |
using bc::nn::Layer_Output_Base< OutputTensorDescriptor >::output_tensor_dim = typename OutputTensorDescriptor::tensor_dim |
using bc::nn::Layer_Output_Base< OutputTensorDescriptor >::output_tensor_dim = typename OutputTensorDescriptor::tensor_dim |
using bc::nn::Layer_Output_Base< OutputTensorDescriptor >::output_tensor_type = typename OutputTensorDescriptor::tensor_type |
using bc::nn::Layer_Output_Base< OutputTensorDescriptor >::output_tensor_type = typename OutputTensorDescriptor::type |
using bc::nn::Layer_Output_Base< OutputTensorDescriptor >::output_value_type = typename OutputTensorDescriptor::value_type |
using bc::nn::Layer_Output_Base< OutputTensorDescriptor >::output_value_type = typename OutputTensorDescriptor::value_type |
|
inlinevirtual |
|
inlinevirtual |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
protected |
|
protected |