25 const string& =
"dense_relu_layer");
38 const string& =
"dense_relu_layer");
48 Index output_features = 0;
52 void configure_operators();
Shape get_output_shape() const override
Returns the layer output shape, derived from input shape and configured output features.
void set_input_shape(const Shape &) override
Sets the input shape; subclasses override to derive dependent dimensions.
void set_output_shape(const Shape &) override
Sets the output shape; subclasses override when the output is user-configurable.
Index get_sequence_length() const
Definition dense_relu_layer.h:31
DenseRelu(const Shape &={}, const Shape &={}, const string &="dense_relu_layer")
Constructs a fused dense+ReLU layer with the given input and output shapes.
void set(const Shape &={}, const Shape &={}, const string &="dense_relu_layer")
Reconfigures the layer with new input/output shapes and label.
void on_compute_dtype_changed() override
Subclass hook invoked when the compute dtype changes; default is no-op.
Definition dense_relu_layer.h:44
Index get_input_features() const
Definition dense_relu_layer.h:30
ActivationOp::Function get_output_activation() const override
Returns the layer's output activation (Identity for most layers; overridden by Dense/Bounding).
Definition dense_relu_layer.h:33
Shape input_shape
Definition layer.h:256
Definition adaptive_moment_estimation.h:14
Function
Supported activation functions.
Definition operators.h:171
@ ReLU
Definition operators.h:171
Fused affine + ReLU activation (uses cuBLASLt epilogue on GPU when available).
Definition operators.h:286
Fixed-capacity small-vector describing tensor dimensions (rank up to MaxRank).
Definition tensor_utilities.h:42