| 96 | } |
| 97 | |
| 98 | core::TensorValue concat_last_dim( |
| 99 | core::ModuleBuildContext & ctx, |
| 100 | const core::TensorValue & lhs, |
| 101 | const core::TensorValue & rhs) { |
| 102 | return ConcatModule({static_cast<int>(lhs.shape.rank - 1)}).build(ctx, lhs, rhs); |
| 103 | } |
| 104 | |
| 105 | core::TensorValue variance_rms_norm( |
| 106 | core::ModuleBuildContext & ctx, |
no test coverage detected