Mila
Deep Neural Network Library
|
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 > > > ¶meters, std::vector< std::shared_ptr< Tensor< TPrecision, MR > > > ¶meter_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 override | Mila::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 > > > ¶meters, std::vector< std::shared_ptr< Tensor< DeviceType::Cuda, MR > > > ¶meter_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) const | Mila::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 > > > ¶meters, const OperationAttributes &properties, Tensor< TPrecision, MR > &losses, std::vector< std::shared_ptr< Tensor< TPrecision, MR > > > &output_state) const override | Mila::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 > > > ¶meters, const OperationAttributes &attributes, Tensor< DeviceType::Cuda, MR > &output, std::vector< std::shared_ptr< Tensor< DeviceType::Cuda, MR > > > &output_state) const=0 | Mila::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() const | Mila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput > | inline |
getDeviceType() const | Mila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput > | inline |
getName() const override | Mila::Dnn::Compute::FusedSoftmaxCrossEntropyOp< TPrecision > | inlinevirtual |
getOperationType() const | Mila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput > | inline |
MR typedef | Mila::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()=default | Mila::Dnn::Compute::BinaryOperation< TPrecision, int, TPrecision, DeviceType::Cuda > | virtual |
~OperationBase()=default | Mila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput > | virtual |