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:
Thomas Santerre
2024-03-18 13:50:14 -04:00
committed by GitHub
parent 58605252e8
commit 04a61a9c72
5 changed files with 236 additions and 20 deletions

View File

@ -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,