Home
last modified time | relevance | path

Searched refs:reshape_to_matrix (Results 1 – 1 of 1) sorted by relevance

/test/mlts/benchmark/tools/
Dtensor_utils.py30 def reshape_to_matrix(array): function
373 reshaped_diff = reshape_to_matrix(self.tensor_dict.calc_diff(layer, relative_error=False))
374 reshaped_cpu = reshape_to_matrix(self.tensor_dict['cpu'][layer])
375 reshaped_nnapi = reshape_to_matrix(self.tensor_dict['nnapi'][layer])
407 reshaped_target_diff = reshape_to_matrix(target_diff)
481 reshaped_diff = reshape_to_matrix(