Lines Matching refs:axs

357   def update_heatmap_data(self, i, fig, axs):  argument
366     for ax_tuple in axs:
369     axs[0][0].set_title('Diff')
370     axs[0][1].set_title('CPU Tensor')
371     axs[0][2].set_title('NNAPI Tensor')
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])
388     axs = []
390       axs.append([])
392           axs[-1].append(fig.add_subplot(spec[row, col]))
395                                   fargs=(fig, axs),
462   def update_hist_comparison_data(self, i, fig, axs, bins=50):  argument
470     for row in axs:
474     hist_ax = axs[0][0]
478     for i, ax in enumerate(axs[1]):
480       axs[1][i].set_title(labels[i])
483       sns.heatmap(reshaped_diff, cmap=cmap, cbar=True, ax=axs[1][i], cbar_ax=axs[2][i],
496     axs = [[], [], []]
497     axs[0].append(fig.add_subplot(gs[0, :]))
500       axs[1].append(fig.add_subplot(gs[1, i]))
502       axs[2].append(fig.add_subplot(gs[2, i]))
504                                   fargs=(fig, axs),