src/transformers/models/efficientnet/modeling_efficientnet.py/EfficientNetExpansionLayer/forward
class EfficientNetExpansionLayer:    def forward(self, hidden_states: torch.FloatTensor) -> torch.Tensor:
        # Expand phase
        hidden_states = self.expand_conv(hidden_states)
        hidden_states = self.expand_bn(hidden_states)
        hidden_states = self.expand_act(hidden_states)

        return hidden_states