55b557f17385ca10b8a3e8fb8bbb0d3799906db5,src/sdk/pynni/nni/compression/tensorflow/builtin_pruners.py,FPGMPruner,_get_min_gm_kernel_idx,#FPGMPruner#Any#Any#,177

Before Change

                dist_list.append(dist_sum)
                idx_list.append([in_i, out_i])
        dist_tensor = tf.convert_to_tensor(dist_list)
        idx_tensor = tf.constant(idx_list)

        _, idx = tf.math.top_k(dist_tensor, k=n)
        return tf.gather(idx_tensor, idx)

After Change

        for in_i in range(weight.shape[0]):
            for out_i in range(weight.shape[1]):
                dist_sum = self._get_distance_sum(weight, in_i, out_i)
                dist_list.append((dist_sum, (in_i, out_i)))
        min_gm_kernels = sorted(dist_list, key=lambda x: x[0])[:n]
        return [x[1] for x in min_gm_kernels]

Instances