Searched refs:weight_name (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/model_pruning/python/ |
| D | pruning.py | 348 weight_name, sparsity = val.split(':')
351 weight_sparsity_map[weight_name] = float(sparsity)
355 def _get_sparsity(self, weight_name): argument
359 if weight_name.find(name) != -1
366 'Multiple matches in weight_sparsity_map for weight %s' % weight_name)
|
| /external/tensorflow/tensorflow/python/keras/saving/ |
| D | hdf5_format.py | 670 return [weights_group[weight_name] for weight_name in optimizer_weight_names]
747 weight_values = [np.asarray(g[weight_name]) for weight_name in weight_names]
803 weight_values = [np.asarray(g[weight_name]) for weight_name in weight_names]
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | callbacks.py | 1366 weight_name = weight.name.replace(':', '_')
1369 summary_ops_v2.histogram(weight_name, weight, step=epoch)
1371 self._log_weight_as_image(weight, weight_name, epoch)
1374 def _log_weight_as_image(self, weight, weight_name, epoch): argument
1396 summary_ops_v2.image(weight_name, w_img, step=epoch)
|