|
ROOT
Reference Guide |
|
Go to the documentation of this file.
18 #ifndef TMVA_DNN_CNN_DESCRIPTORS
19 #define TMVA_DNN_CNN_DESCRIPTORS
36 template <
typename Layer_t>
50 template <
typename Layer_t>
61 template <
typename Layer_t>
94 template <
typename Architecture_t>
109 std::vector<TensorDescriptor_t>
xDesc;
110 std::vector<TensorDescriptor_t>
yDesc;
115 template <
typename Layer_t>
size_t * BackwardWorkspace
WeightsDescriptor_t WeightsGradDescriptor
std::vector< TensorDescriptor_t > dyDesc
typename Layer_t::LayerDescriptor_t LayerDescriptor_t
AlgorithmHelper_t HelperAlgorithm
HelperDescriptor_t HelperDescriptor
WeightsDescriptor_t WeightsDescriptor
HelperDescriptor_t HelperDescriptor
size_t fReductionWorkspaceSize
AlgorithmBackward_t AlgorithmBackward
typename Architecture_t::DropoutDescriptor_t HelperDescriptor_t
typename Layer_t::AlgorithmForward_t AlgorithmForward_t
WeightsDescriptor_t WeightsDescriptor
typename Layer_t::AlgorithmHelper_t AlgorithmHelper_t
void * fReductionWorkspace
typename Layer_t::ReduceTensorDescriptor_t ReduceTensorDescriptor_t
size_t HelperWorkspaceSize
typename Architecture_t::TensorDescriptor_t TensorDescriptor_t
typename Layer_t::HelperDescriptor_t HelperDescriptor_t
std::vector< TensorDescriptor_t > yDesc
size_t * ForwardWorkspace
std::vector< TensorDescriptor_t > xDesc
std::vector< TensorDescriptor_t > dxDesc
LayerDescriptor_t LayerDescriptor
ReduceTensorDescriptor_t fReduceTensorDesc
HelperDescriptor_t HelperDescriptor
typename Architecture_t::RecurrentDescriptor_t LayerDescriptor_t
typename Layer_t::AlgorithmDataType_t AlgorithmDataType_t
typename Architecture_t::FilterDescriptor_t WeightsDescriptor_t
size_t ForwardWorkspaceSize
typename Layer_t::AlgorithmBackward_t AlgorithmBackward_t
size_t HelperWorkspaceSize
typename Layer_t::WeightsDescriptor_t WeightsDescriptor_t
AlgorithmDataType_t DataType
LayerDescriptor_t LayerDescriptor
typename Layer_t::HelperDescriptor_t HelperDescriptor_t
size_t BackwardWorkspaceSize
size_t ForwardWorkspaceSize
AlgorithmForward_t AlgorithmForward
create variable transformations