Searched refs:adaptive_max_pooling2d_fn (Results 1 – 1 of 1) sorted by relevance
16 using adaptive_max_pooling2d_fn = void(*)(const Tensor& output, const Tensor& indices, const Tensor… variable18 DECLARE_DISPATCH(adaptive_max_pooling2d_fn, adaptive_max_pool2d_kernel);