48 #ifndef __NEURAL_NENTWORK_LOSS_SOFTMAX_CROSS_LAYER_H__
49 #define __NEURAL_NENTWORK_LOSS_SOFTMAX_CROSS_LAYER_H__
51 #include "algorithms/algorithm.h"
52 #include "data_management/data/tensor.h"
53 #include "services/daal_defines.h"
54 #include "algorithms/neural_networks/layers/layer.h"
55 #include "algorithms/neural_networks/layers/loss/loss_layer.h"
56 #include "algorithms/neural_networks/layers/loss/softmax_cross_layer_types.h"
57 #include "algorithms/neural_networks/layers/loss/softmax_cross_layer_forward.h"
58 #include "algorithms/neural_networks/layers/loss/softmax_cross_layer_backward.h"
64 namespace neural_networks
73 namespace softmax_cross
97 template<
typename algorithmFPType = DAAL_ALGORITHM_FP_TYPE, Method method = defaultDense>
98 class Batch :
public loss::Batch
104 forward::Batch<algorithmFPType, method> *forwardLayerObject =
new forward::Batch<algorithmFPType, method>(parameter);
105 backward::Batch<algorithmFPType, method> *backwardLayerObject =
new backward::Batch<algorithmFPType, method>(parameter);
107 LayerIface::forwardLayer = services::SharedPtr<forward::Batch<algorithmFPType, method> >(forwardLayerObject);
108 LayerIface::backwardLayer = services::SharedPtr<backward::Batch<algorithmFPType, method> >(backwardLayerObject);
115 using interface1::Batch;
daal
Definition: algorithm_base_common.h:57
daal::algorithms::neural_networks::layers::loss::softmax_cross::interface1::Batch
Provides methods for the softmax cross-entropy layer in the batch processing mode.
Definition: softmax_cross_layer.h:98
daal::algorithms::neural_networks::layers::loss::softmax_cross::interface1::Batch::Batch
Batch()
Definition: softmax_cross_layer.h:102
daal::algorithms::neural_networks::layers::loss::interface1::Batch
Provides methods for the loss layer in the batch processing mode.
Definition: loss_layer.h:92
daal::algorithms::neural_networks::layers::loss::softmax_cross::interface1::Batch::parameter
Parameter parameter
Definition: softmax_cross_layer.h:109
daal::algorithms::neural_networks::layers::loss::softmax_cross::interface1::Parameter
Parameters for the softmax cross-entropy layer.
Definition: softmax_cross_layer_types.h:108