Searched refs:WALSMatrixFactorization (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| D | wals.py | 128 WALSMatrixFactorization.COMPLETED_SWEEPS + ":0")
130 raise RuntimeError(WALSMatrixFactorization.COMPLETED_SWEEPS +
181 input_rows = features[WALSMatrixFactorization.INPUT_ROWS]
182 input_cols = features[WALSMatrixFactorization.INPUT_COLS]
211 name=WALSMatrixFactorization.COMPLETED_SWEEPS,
216 name=WALSMatrixFactorization.LOSS,
223 name=WALSMatrixFactorization.RWSE,
320 WALSMatrixFactorization.PROJECTION_WEIGHTS)
335 WALSMatrixFactorization.PROJECTION_RESULT: control_flow_ops.cond(
336 features[WALSMatrixFactorization.PROJECT_ROW],
[all …]
|
| D | wals_test.py | 136 wals_lib.WALSMatrixFactorization.INPUT_ROWS: rows,
137 wals_lib.WALSMatrixFactorization.INPUT_COLS: cols,
165 features[wals_lib.WALSMatrixFactorization.PROJECT_ROW] = (
174 features[wals_lib.WALSMatrixFactorization.PROJECTION_WEIGHTS] = (
243 self._model = wals_lib.WALSMatrixFactorization(
417 self._model = wals_lib.WALSMatrixFactorization(1, 1, 1, config=config)
490 8, name=wals_lib.WALSMatrixFactorization.COMPLETED_SWEEPS)
|