Lines Matching refs:ax
309 def __sns_distplot(self, layer, bins, ax, range, relative_error): argument
311 hist_kws={"range":range, "log":True}, ax=ax, kde=False)
313 def __plt_hist(self, layer, bins, ax, range, relative_error): argument
314 ax.hist(self.tensor_dict.calc_diff(layer, relative_error=relative_error), bins=bins,
329 for ax in (ax1, ax2):
330 ax.clear()
337 hist_func(layer=layer, bins=bins, ax=ax1,
339 hist_func(layer=layer, bins=bins, ax=ax2,
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,
367 for ax in ax_tuple:
368 ax.clear()
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])
408 fig, ax = subplots(figsize=(9, 9))
471 for ax in row:
472 ax.clear()
478 for i, ax in enumerate(axs[1]):
483 sns.heatmap(reshaped_diff, cmap=cmap, cbar=True, ax=axs[1][i], cbar_ax=axs[2][i],
486 hist_kws={"log":True}, ax=hist_ax, kde=False)