| asLeftTensor(const ITensor &t) | Mila::Dnn::Compute::BinaryOperation< DeviceType::Cuda, TPrecision, TPrecision, TensorDataType::INT32 > | inlineprotectedstatic |
| asOutputTensor(ITensor &t) | Mila::Dnn::Compute::BinaryOperation< DeviceType::Cuda, TPrecision, TPrecision, TensorDataType::INT32 > | inlineprotectedstatic |
| asRightTensor(const ITensor &t) | Mila::Dnn::Compute::BinaryOperation< DeviceType::Cuda, TPrecision, TPrecision, TensorDataType::INT32 > | inlineprotectedstatic |
| backward(const ITensor &logits, const ITensor &targets, const ITensor &loss_grad, ITensor &logits_grad, ITensor &targets_grad) const override | Mila::Dnn::Compute::Cuda::SoftmaxCrossEntropy::CudaSoftmaxCrossEntropyOp< TPrecision, TLogits, TTargets > | inlinevirtual |
| BinaryOperationBase typedef | Mila::Dnn::Compute::Cuda::SoftmaxCrossEntropy::CudaSoftmaxCrossEntropyOp< TPrecision, TLogits, TTargets > | |
| build(const BuildContext &config) override | Mila::Dnn::Compute::Cuda::SoftmaxCrossEntropy::CudaSoftmaxCrossEntropyOp< TPrecision, TLogits, TTargets > | inlinevirtual |
| cached_batch_size_ | Mila::Dnn::Compute::Cuda::SoftmaxCrossEntropy::CudaSoftmaxCrossEntropyOp< TPrecision, TLogits, TTargets > | private |
| cached_probs_ | Mila::Dnn::Compute::Cuda::SoftmaxCrossEntropy::CudaSoftmaxCrossEntropyOp< TPrecision, TLogits, TTargets > | mutableprivate |
| cached_seq_len_ | Mila::Dnn::Compute::Cuda::SoftmaxCrossEntropy::CudaSoftmaxCrossEntropyOp< TPrecision, TLogits, TTargets > | private |
| cached_stream_ | Mila::Dnn::Compute::Cuda::SoftmaxCrossEntropy::CudaSoftmaxCrossEntropyOp< TPrecision, TLogits, TTargets > | private |
| cached_vocab_size_ | Mila::Dnn::Compute::Cuda::SoftmaxCrossEntropy::CudaSoftmaxCrossEntropyOp< TPrecision, TLogits, TTargets > | private |
| class_weights_ | Mila::Dnn::Compute::Cuda::SoftmaxCrossEntropy::CudaSoftmaxCrossEntropyOp< TPrecision, TLogits, TTargets > | private |
| clearGradients() noexcept | Mila::Dnn::Compute::Operation< TDeviceType, TPrecision > | inlinevirtual |
| config_ | Mila::Dnn::Compute::Cuda::SoftmaxCrossEntropy::CudaSoftmaxCrossEntropyOp< TPrecision, TLogits, TTargets > | private |
| context_ | Mila::Dnn::Compute::Cuda::SoftmaxCrossEntropy::CudaSoftmaxCrossEntropyOp< TPrecision, TLogits, TTargets > | private |
| CudaExecutionContext typedef | Mila::Dnn::Compute::Cuda::SoftmaxCrossEntropy::CudaSoftmaxCrossEntropyOp< TPrecision, TLogits, TTargets > | |
| CudaSoftmaxCrossEntropyOp(IExecutionContext *context, const CrossEntropyConfig &config) | Mila::Dnn::Compute::Cuda::SoftmaxCrossEntropy::CudaSoftmaxCrossEntropyOp< TPrecision, TLogits, TTargets > | inline |
| data_type | Mila::Dnn::Compute::Operation< TDeviceType, TPrecision > | static |
| DataTypeTraits typedef | Mila::Dnn::Compute::Operation< TDeviceType, TPrecision > | |
| device_type | Mila::Dnn::Compute::Operation< TDeviceType, TPrecision > | static |
| forward(const ITensor &logits, const ITensor &targets, ITensor &output) const override | Mila::Dnn::Compute::Cuda::SoftmaxCrossEntropy::CudaSoftmaxCrossEntropyOp< TPrecision, TLogits, TTargets > | inlinevirtual |
| getConfig() const | Mila::Dnn::Compute::Cuda::SoftmaxCrossEntropy::CudaSoftmaxCrossEntropyOp< TPrecision, TLogits, TTargets > | inline |
| getDataType() const | Mila::Dnn::Compute::Operation< TDeviceType, TPrecision > | inlinevirtual |
| getDeviceType() const | Mila::Dnn::Compute::Operation< TDeviceType, TPrecision > | inlinevirtual |
| getName() const override | Mila::Dnn::Compute::Cuda::SoftmaxCrossEntropy::CudaSoftmaxCrossEntropyOp< TPrecision, TLogits, TTargets > | inlinevirtual |
| getOperationType() const override | Mila::Dnn::Compute::Cuda::SoftmaxCrossEntropy::CudaSoftmaxCrossEntropyOp< TPrecision, TLogits, TTargets > | inlinevirtual |
| getStateMemorySize() const | Mila::Dnn::Compute::Operation< TDeviceType, TPrecision > | inlinevirtual |
| is_built_ | Mila::Dnn::Compute::Operation< TDeviceType, TPrecision > | protected |
| isBuilt() const | Mila::Dnn::Compute::Operation< TDeviceType, TPrecision > | inlinevirtual |
| isEvalMode() const | Mila::Dnn::Compute::Operation< TDeviceType, TPrecision > | inlinevirtual |
| LogitsTensorType typedef | Mila::Dnn::Compute::Cuda::SoftmaxCrossEntropy::CudaSoftmaxCrossEntropyOp< TPrecision, TLogits, TTargets > | |
| MR typedef | Mila::Dnn::Compute::Cuda::SoftmaxCrossEntropy::CudaSoftmaxCrossEntropyOp< TPrecision, TLogits, TTargets > | |
| NativeType typedef | Mila::Dnn::Compute::Cuda::SoftmaxCrossEntropy::CudaSoftmaxCrossEntropyOp< TPrecision, TLogits, TTargets > | |
| ParameterGradTensor typedef | Mila::Dnn::Compute::BinaryOperation< DeviceType::Cuda, TPrecision, TPrecision, TensorDataType::INT32 > | |
| ParameterTensor typedef | Mila::Dnn::Compute::BinaryOperation< DeviceType::Cuda, TPrecision, TPrecision, TensorDataType::INT32 > | |
| setGradients(ITensor *weight_grad, ITensor *bias_grad) | Mila::Dnn::Compute::Operation< TDeviceType, TPrecision > | inlinevirtual |
| setParameters(ITensor *class_weights, ITensor *bias) override | Mila::Dnn::Compute::Cuda::SoftmaxCrossEntropy::CudaSoftmaxCrossEntropyOp< TPrecision, TLogits, TTargets > | inlinevirtual |
| setTrainingMode(TrainingMode training_mode) | Mila::Dnn::Compute::Operation< TDeviceType, TPrecision > | inlinevirtual |
| TargetsNativeType typedef | Mila::Dnn::Compute::Cuda::SoftmaxCrossEntropy::CudaSoftmaxCrossEntropyOp< TPrecision, TLogits, TTargets > | |
| TargetsTensorType typedef | Mila::Dnn::Compute::Cuda::SoftmaxCrossEntropy::CudaSoftmaxCrossEntropyOp< TPrecision, TLogits, TTargets > | |
| TensorLeftType typedef | Mila::Dnn::Compute::BinaryOperation< DeviceType::Cuda, TPrecision, TPrecision, TensorDataType::INT32 > | |
| TensorOutputType typedef | Mila::Dnn::Compute::BinaryOperation< DeviceType::Cuda, TPrecision, TPrecision, TensorDataType::INT32 > | |
| TensorRightType typedef | Mila::Dnn::Compute::BinaryOperation< DeviceType::Cuda, TPrecision, TPrecision, TensorDataType::INT32 > | |
| training_mode_ | Mila::Dnn::Compute::Operation< TDeviceType, TPrecision > | protected |
| ~BinaryOperation()=default | Mila::Dnn::Compute::BinaryOperation< DeviceType::Cuda, TPrecision, TPrecision, TensorDataType::INT32 > | virtual |
| ~Operation()=default | Mila::Dnn::Compute::Operation< TDeviceType, TPrecision > | virtual |
1.15.0