Executes the worker's pipeline for this tensor.
(
&mut self,
x: &Tensor,
index_pos: usize,
block_idx: usize,
_: &mut Context,
)
| 145 | |
| 146 | /// Executes the worker's pipeline for this tensor. |
| 147 | async fn forward_mut( |
| 148 | &mut self, |
| 149 | x: &Tensor, |
| 150 | index_pos: usize, |
| 151 | block_idx: usize, |
| 152 | _: &mut Context, |
| 153 | ) -> Result<Tensor> { |
| 154 | log::debug!("forwarding single op"); |
| 155 | self.forward_request(Message::single_op( |
| 156 | &self.layer_name, |
| 157 | x, |
| 158 | index_pos, |
| 159 | block_idx, |
| 160 | )) |
| 161 | .await |
| 162 | } |
| 163 | |
| 164 | /// Executes the worker's pipeline with multiple batched steps for this tensor. |
| 165 | async fn forward_batch( |
nothing calls this directly
no test coverage detected