(
&mut self,
x: &Tensor,
index_pos: usize,
block_idx: usize,
ctx: &mut Context,
)
| 75 | } |
| 76 | |
| 77 | async fn forward_mut( |
| 78 | &mut self, |
| 79 | x: &Tensor, |
| 80 | index_pos: usize, |
| 81 | block_idx: usize, |
| 82 | ctx: &mut Context, |
| 83 | ) -> anyhow::Result<Tensor> { |
| 84 | self.forward(x, index_pos, block_idx, ctx).await |
| 85 | } |
| 86 | |
| 87 | fn layer_name(&self) -> &str { |
| 88 | self.layer_name |