Searched refs:axs (Results 1 – 1 of 1) sorted by relevance
| /test/mlts/benchmark/tools/ |
| D | tensor_utils.py | 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([])
[all …]
|