Mila
Deep Neural Network Library
Loading...
Searching...
No Matches
Mila::Dnn::Compute::FusedSoftmaxCrossEntropyOp< TPrecision > Member List

This is the complete list of members for Mila::Dnn::Compute::FusedSoftmaxCrossEntropyOp< TPrecision >, including all inherited members.

backward(const Tensor< TPrecision, MR > &input1, const Tensor< int, MR > &input2, const Tensor< TPrecision, MR > &output, const Tensor< TPrecision, MR > &output_gradient, const std::vector< std::shared_ptr< Tensor< TPrecision, MR > > > &parameters, std::vector< std::shared_ptr< Tensor< TPrecision, MR > > > &parameter_gradients, Tensor< TPrecision, MR > &input1_gradient, Tensor< int, MR > &input2_gradient, const OperationAttributes &properties, const std::vector< std::shared_ptr< Tensor< TPrecision, MR > > > &output_state) const overrideMila::Dnn::Compute::FusedSoftmaxCrossEntropyOp< TPrecision >inline
BinaryOperation< TPrecision, int, TPrecision, DeviceType::Cuda >::backward(const Tensor< int, MR > &input1, const Tensor< TPrecision, MR > &input2, const Tensor< DeviceType::Cuda, MR > &output, const Tensor< DeviceType::Cuda, MR > &output_gradient, const std::vector< std::shared_ptr< Tensor< int, MR > > > &parameters, std::vector< std::shared_ptr< Tensor< DeviceType::Cuda, MR > > > &parameter_gradients, Tensor< int, MR > &input1_gradient, Tensor< TPrecision, MR > &input2_gradient, const OperationAttributes &attributes, const std::vector< std::shared_ptr< Tensor< DeviceType::Cuda, MR > > > &output_state) constMila::Dnn::Compute::BinaryOperation< TPrecision, int, TPrecision, DeviceType::Cuda >inlinevirtual
BinaryOperation(OperationType operation_type)Mila::Dnn::Compute::BinaryOperation< TPrecision, int, TPrecision, DeviceType::Cuda >inline
BinaryOperation(OperationType operation_type, std::shared_ptr< DeviceContext > context)Mila::Dnn::Compute::BinaryOperation< TPrecision, int, TPrecision, DeviceType::Cuda >inline
device_context_Mila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput >private
forward(const Tensor< TPrecision, MR > &logits, const Tensor< int, MR > &targets, const std::vector< std::shared_ptr< Tensor< TPrecision, MR > > > &parameters, const OperationAttributes &properties, Tensor< TPrecision, MR > &losses, std::vector< std::shared_ptr< Tensor< TPrecision, MR > > > &output_state) const overrideMila::Dnn::Compute::FusedSoftmaxCrossEntropyOp< TPrecision >inline
BinaryOperation< TPrecision, int, TPrecision, DeviceType::Cuda >::forward(const Tensor< int, MR > &input1, const Tensor< TPrecision, MR > &input2, const std::vector< std::shared_ptr< Tensor< int, MR > > > &parameters, const OperationAttributes &attributes, Tensor< DeviceType::Cuda, MR > &output, std::vector< std::shared_ptr< Tensor< DeviceType::Cuda, MR > > > &output_state) const=0Mila::Dnn::Compute::BinaryOperation< TPrecision, int, TPrecision, DeviceType::Cuda >pure virtual
FusedSoftmaxCrossEntropyOp()Mila::Dnn::Compute::FusedSoftmaxCrossEntropyOp< TPrecision >inline
FusedSoftmaxCrossEntropyOp(std::shared_ptr< DeviceContext > context)Mila::Dnn::Compute::FusedSoftmaxCrossEntropyOp< TPrecision >inline
getDeviceContext() constMila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput >inline
getDeviceType() constMila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput >inline
getName() const overrideMila::Dnn::Compute::FusedSoftmaxCrossEntropyOp< TPrecision >inlinevirtual
getOperationType() constMila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput >inline
MR typedefMila::Dnn::Compute::FusedSoftmaxCrossEntropyOp< TPrecision >
operation_type_Mila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput >private
OperationBase(OperationType operation_type, std::shared_ptr< DeviceContext > context)Mila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput >inline
~BinaryOperation()=defaultMila::Dnn::Compute::BinaryOperation< TPrecision, int, TPrecision, DeviceType::Cuda >virtual
~OperationBase()=defaultMila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput >virtual