Searched defs:adaptive_avg_pool2d (Results 1 – 8 of 8) sorted by relevance
568 inline Tensor adaptive_avg_pool2d( in adaptive_avg_pool2d() function591 inline Tensor adaptive_avg_pool2d( in adaptive_avg_pool2d() function
13 Tensor adaptive_avg_pool2d( in adaptive_avg_pool2d() function
133 def adaptive_avg_pool2d(input: Tensor, output_size: BroadcastingList2[int]) -> Tensor: function
830 auto adaptive_avg_pool2d = getInputTensorQParamOpFusionInfo( in quant_fusion_pattern_and_replacements() local
59 def adaptive_avg_pool2d(self: List[int], out: List[int]): function
1370 def adaptive_avg_pool2d(input: Tensor, output_size: BroadcastingList2[int]) -> Tensor: function
2466 def adaptive_avg_pool2d(input: Tensor, output_size: Tuple[int, int]): function
416 TEST_F(VulkanAPITest, adaptive_avg_pool2d) { in TEST_F() argument