(
&mut self,
x: &Tensor,
index_pos: usize,
block_idx: usize,
ctx: &mut Context,
)
| 135 | } |
| 136 | |
| 137 | async fn forward_mut( |
| 138 | &mut self, |
| 139 | x: &Tensor, |
| 140 | index_pos: usize, |
| 141 | block_idx: usize, |
| 142 | ctx: &mut Context, |
| 143 | ) -> Result<Tensor> { |
| 144 | self.forward(x, index_pos, block_idx, ctx).await |
| 145 | } |
| 146 | |
| 147 | fn layer_name(&self) -> &str { |
| 148 | &self.name |