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

Before Change

            for out_i in range(weight.shape[1]):
                dist_sum = self._get_distance_sum(weight, in_i, out_i)
                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)

After Change

            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]

    def _get_distance_sum(self, weight, in_idx, out_idx):
        w = tf.reshape(weight, (-1, weight.shape[-2], weight.shape[-1]))

Instances