Searched refs:adaptive_max_pooling3d_fn (Results 1 – 1 of 1) sorted by relevance
26 using adaptive_max_pooling3d_fn = void(*)(const Tensor& output, const Tensor& indices, const Tensor… variable28 DECLARE_DISPATCH(adaptive_max_pooling3d_fn, adaptive_max_pool3d_kernel);