Searched refs:cbar_ax (Results 1 – 1 of 1) sorted by relevance
| /test/mlts/benchmark/tools/ |
| D | tensor_utils.py | 353 def __sns_heatmap(self, data, ax, cbar_ax, **kwargs): argument
354 return sns.heatmap(data, cmap=self.cmap, cbar=True, ax=ax, cbar_ax=cbar_ax,
377 g1 = self.__sns_heatmap(data=reshaped_diff, ax=axs[0][0], cbar_ax=axs[1][0],
379 g2 = self.__sns_heatmap(data=reshaped_cpu, ax=axs[0][1], cbar_ax=axs[1][1])
380 g3 = self.__sns_heatmap(data=reshaped_nnapi, ax=axs[0][2], cbar_ax=axs[1][2])
483 sns.heatmap(reshaped_diff, cmap=cmap, cbar=True, ax=axs[1][i], cbar_ax=axs[2][i],
|