Searched refs:_low_memory_max_pool2d_with_offsets (Results 1 – 4 of 4) sorted by relevance
168 _low_memory_max_pool2d_with_offsets = make_prim( variable
965 vals, offsets = prims._low_memory_max_pool2d_with_offsets(
3809 @register_lowering(prims._low_memory_max_pool2d_with_offsets, type_promotion_kind=None)3810 def _low_memory_max_pool2d_with_offsets( function4311 v, offsets = _low_memory_max_pool2d_with_offsets(
1315 prims._low_memory_max_pool2d_with_offsets.default,