Searched refs:target_layer (Results 1 – 1 of 1) sorted by relevance
254 def plot_error_heatmap(self, target_layer, length=1): argument255 target_diff = self.tensor_dict['cpu'][target_layer] - \256 self.tensor_dict['nnapi'][target_layer]