CrossEntropyConfig.ixx File Reference

Configuration for the fused SoftmaxCrossEntropy loss module. More...

#include <stdexcept>
#include <cstdint>
#include <utility>
#include <string>
#include <ostream>
#include <sstream>
import Serialization.Metadata;
import Dnn.ComponentConfig;
import Dnn.Component;

Classes
class	Mila::Dnn::CrossEntropyConfig
	Configuration for fused SoftmaxCrossEntropy loss. More...

Namespaces
namespace	Mila
	Mila main API namespace.
namespace	Mila::Dnn

Detailed Description

Configuration for the fused SoftmaxCrossEntropy loss module.

Minimal configuration used by the fused softmax + cross-entropy kernels.