LBANN  0.103.0
LivermoreBigArtificialNeuralNetworkToolkit
lbann::openmp_backend Member List

This is the complete list of members for lbann::openmp_backend, including all inherited members.

data_type()lbann::openmp_backendinlinestatic
devicelbann::openmp_backendstatic
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_backendinlinestatic
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_backendinlinestatic
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_backendstatic
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_backendstatic
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