Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/factorization/python/ops/
Dfactorization_ops_test.py55 wals_model.row_factors,
206 row_factors = [x.eval() for x in wals_model.row_factors]
208 self.assertAllClose(row_factors[0], self._row_factors_0, atol=1e-3)
209 self.assertAllClose(row_factors[1], self._row_factors_1, atol=1e-3)
381 row_factors = [x.eval() for x in wals_model.row_factors]
383 self.assertAllClose(row_factors[0], self._row_factors_0, atol=1e-3)
384 self.assertAllClose(row_factors[1], self._row_factors_1, atol=1e-3)
526 row_factors1 = [x.eval() for x in als_model.row_factors]
546 row_factors2 = [x.eval() for x in wals_model.row_factors]
627 als_model.row_factors[0].eval()[1], als_model.row_factors[0].eval()[3]
[all …]
Dfactorization_ops_test_utils.py98 def calculate_loss(input_mat, row_factors, col_factors, regularization=None, argument
128 gathered_row_factors = array_ops.gather(row_factors, row_indices)
138 row_norm = math_ops.reduce_sum(math_ops.square(row_factors))
141 row_factors, col_factors, transpose_b=True)))
Dwals_test.py265 row_factors = self._model.get_row_factors()
266 self.assertAllClose(row_factors[0], self._row_factors_0, atol=1e-3)
267 self.assertAllClose(row_factors[1], self._row_factors_1, atol=1e-3)
Dfactorization_ops.py299 def row_factors(self): member in WALSModel