Searched refs:target_layer (Results 1 – 1 of 1) sorted by relevance
403 def plot_error_heatmap(self, target_layer, vmin=None, vmax=None): argument405 target_diff = self.tensor_dict['cpu'][target_layer] - \406 self.tensor_dict['nnapi'][target_layer]