Searched refs:sns (Results 1 – 1 of 1) sorted by relevance
| /test/mlts/benchmark/tools/ |
| D | tensor_utils.py | 14 import seaborn as sns namespace
301 self.cmap = sns.diverging_palette(220, 20, sep=20, as_cmap=True)
310 sns.distplot(self.tensor_dict.calc_diff(layer, relative_error=relative_error), bins=bins,
354 return sns.heatmap(data, cmap=self.cmap, cbar=True, ax=ax, cbar_ax=cbar_ax,
410 sns.heatmap(reshaped_target_diff,
477 cmap = sns.diverging_palette(220, 20, sep=20, as_cmap=True)
483 sns.heatmap(reshaped_diff, cmap=cmap, cbar=True, ax=axs[1][i], cbar_ax=axs[2][i],
485 sns.distplot(model_data.tensor_dict.calc_diff(layer, relative_error=False), bins=bins,
|