(
forwarder: &mut Box<dyn Forwarder>,
image: Tensor,
ctx: &mut Context,
)
| 106 | } |
| 107 | |
| 108 | pub async fn encode( |
| 109 | forwarder: &mut Box<dyn Forwarder>, |
| 110 | image: Tensor, |
| 111 | ctx: &mut Context, |
| 112 | ) -> anyhow::Result<Tensor> { |
| 113 | let tensors = Vec::from([Tensor::from_slice(&[1f32], 1, &ctx.device)?, image]); |
| 114 | |
| 115 | let combined_tensor = pack_tensors(tensors, &ctx.device)?; |
| 116 | |
| 117 | forwarder.forward_mut(&combined_tensor, 0, 0, ctx).await |
| 118 | } |
| 119 | |
| 120 | /// The constant layer name for this component. |
| 121 | #[cfg(test)] |
no test coverage detected