Searched defs:pooling_fn (Results 1 – 3 of 3) sorted by relevance
93 using pooling_fn = void (*)(Tensor&, const Tensor&, const PoolingParams1D&); variable
19 IntArrayRef padding, IntArrayRef dilation, bool ceil_mode, int64_t n, Func pooling_fn) { in max_pool_with_indices_batch_rule_helper()
4117 def _adaptive_pooling_fn( argument4158 def _adaptive_pooling_fn_with_idx( argument