Searched refs:heatmap (Results 1 – 1 of 1) sorted by relevance
| /test/mlts/benchmark/tools/ |
| D | tensor_utils.py | 135 def generate_animation_html(self, output_file_path, model_names=None, heatmap=True): argument
145 if heatmap:
169 model_names=None, heatmap=True): argument
354 return sns.heatmap(data, cmap=self.cmap, cbar=True, ax=ax, cbar_ax=cbar_ax,
410 sns.heatmap(reshaped_target_diff,
483 sns.heatmap(reshaped_diff, cmap=cmap, cbar=True, ax=axs[1][i], cbar_ax=axs[2][i],
|