(
&mut self,
x: &Tensor,
index_pos: usize,
block_idx: usize,
ctx: &mut Context,
)
| 116 | } |
| 117 | |
| 118 | async fn forward_mut( |
| 119 | &mut self, |
| 120 | x: &Tensor, |
| 121 | index_pos: usize, |
| 122 | block_idx: usize, |
| 123 | ctx: &mut Context, |
| 124 | ) -> Result<Tensor> { |
| 125 | self.forward(x, index_pos, block_idx, ctx).await |
| 126 | } |
| 127 | |
| 128 | fn layer_name(&self) -> &str { |
| 129 | &self.name |