Searched refs:factorized_pool (Results 1 – 3 of 3) sorted by relevance
67 pooled_weights_factorized_pool = pruning_utils.factorized_pool(
164 def factorized_pool(input_tensor, function
456 pool_fn = pruning_utils.factorized_pool