mirror of
https://github.com/huggingface/candle.git
synced 2025-06-20 12:06:35 +00:00
Add avg_pool2d metal implementation for the metal backend (#1869)
* implement metal avg pool 2d * fixX * add suggested precision workaround for the accumulator
This commit is contained in:
@ -1827,7 +1827,7 @@ fn divide(m: usize, b: usize) -> NSUInteger {
|
||||
}
|
||||
|
||||
#[allow(clippy::too_many_arguments)]
|
||||
pub fn call_max_pool2d(
|
||||
pub fn call_pool2d(
|
||||
device: &Device,
|
||||
command_buffer: &CommandBufferRef,
|
||||
kernels: &Kernels,
|
||||
|
Reference in New Issue
Block a user