|
LBANN
0.103.0
LivermoreBigArtificialNeuralNetworkToolkit
|
This is the complete list of members for lbann::openmp_backend, including all inherited members.
| data_type() | lbann::openmp_backend | inlinestatic |
| device | lbann::openmp_backend | static |
| logsoftmax_backward(ScalarT const &alpha_in, TensorDescriptor const &yDesc, El::Matrix< DataT, device > const &y, TensorDescriptor const &dyDesc, El::Matrix< DataT, device > const &dy, ScalarT const &beta_in, TensorDescriptor const &dxDesc, El::Matrix< DataT, device > &dx, El::SyncInfo< device > const &si, softmax_mode mode, softmax_alg alg=softmax_alg::ACCURATE) | lbann::openmp_backend | inlinestatic |
| logsoftmax_forward(ScalarT const &alpha_in, TensorDescriptor const &xDesc, El::Matrix< DataT, device > const &x, ScalarT const &beta_in, TensorDescriptor const &yDesc, El::Matrix< DataT, device > &y, El::SyncInfo< device > const &si, softmax_mode mode) | lbann::openmp_backend | inlinestatic |
| softmax_backward(ScalarT const &alpha_in, TensorDescriptor const &yDesc, El::Matrix< DataT, device > const &y, TensorDescriptor const &dyDesc, El::Matrix< DataT, device > const &dy, ScalarT const &beta_in, TensorDescriptor const &dxDesc, El::Matrix< DataT, device > &dx, El::SyncInfo< device > const &si, softmax_mode mode, softmax_alg alg=softmax_alg::ACCURATE) | lbann::openmp_backend | static |
| softmax_backward(ScalarT const &alpha_in, TensorDescriptor const &outputDesc, El::Matrix< DataT, El::Device::CPU > const &local_output, TensorDescriptor const &outputGradDesc, El::Matrix< DataT, El::Device::CPU > const &local_gradient_wrt_output, ScalarT const &beta_in, TensorDescriptor const &inputGradDesc, El::Matrix< DataT, El::Device::CPU > &local_gradient_wrt_input, El::SyncInfo< El::Device::CPU > const &si, softmax_mode mode, softmax_alg alg) | lbann::openmp_backend | |
| softmax_forward(ScalarT const &alpha_in, TensorDescriptor const &xDesc, El::Matrix< DataT, device > const &x, ScalarT const &beta_in, TensorDescriptor const &yDesc, El::Matrix< DataT, device > &y, El::SyncInfo< device > const &si, softmax_mode mode, softmax_alg alg=softmax_alg::ACCURATE) | lbann::openmp_backend | static |
| softmax_forward(ScalarT const &alpha_in, TensorDescriptor const &inputDesc, El::Matrix< DataT, El::Device::CPU > const &local_input, ScalarT const &beta_in, TensorDescriptor const &outputDesc, El::Matrix< DataT, El::Device::CPU > &local_output, El::SyncInfo< El::Device::CPU > const &si, softmax_mode mode, softmax_alg alg) | lbann::openmp_backend |