Interface LinearOp

All Known Subinterfaces:
MxFp8Layer
All Known Implementing Classes:
Fp8BlockLinear, Fp8BlockMarlinLinear, Fp8Linear, LinearLayer, MarlinLinear, MxFp8BlockLinear, MxFp8MarlinLinear, Nvfp4Linear
Functional Interface:
This is a functional interface and can therefore be used as the assignment target for a lambda expression or method reference.
@FunctionalInterface public interface LinearOp
Shared forward interface for dense, FP8, NVFP4, and Marlin linear layers.

  • Method Summary

    Modifier and Type
    Method
    Description
    Tensor
    forward(Tensor input)
    Applies the linear transformation.

  • Method Details

    • forward

      Tensor forward(Tensor input)
      Applies the linear transformation.
      Parameters:
      input - activations (caller retains ownership).
      Returns:
      owned output tensor (caller must close).