max_matrix_size (reference) in projects: external

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

/external/tensorflow/tensorflow/contrib/opt/python/training/
D	weight_decay_optimizers.py	`395 max_matrix_size=768, argument 441 max_matrix_size=max_matrix_size,`
D	shampoo.py	`73 max_matrix_size=768, argument 125 self._max_matrix_size = max_matrix_size`