Searched refs:INPUT_MATRIX (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| D | factorization_ops_test.py | 34 INPUT_MATRIX = factorization_ops_test_utils.INPUT_MATRIX variable
41 return np_matrix_to_tf_sparse(INPUT_MATRIX)
190 sp_r0 = np_matrix_to_tf_sparse(INPUT_MATRIX, [0, 2]).eval()
191 sp_r1 = np_matrix_to_tf_sparse(INPUT_MATRIX, [1, 4], shuffle=True).eval()
192 sp_r2 = np_matrix_to_tf_sparse(INPUT_MATRIX, [3], shuffle=True).eval()
226 np_matrix_to_tf_sparse(INPUT_MATRIX, [1, 4], shuffle=False)
257 sp_c0 = np_matrix_to_tf_sparse(INPUT_MATRIX, col_slices=[2, 0]).eval()
259 INPUT_MATRIX, col_slices=[5, 3, 1], shuffle=True).eval()
260 sp_c2 = np_matrix_to_tf_sparse(INPUT_MATRIX, col_slices=[4, 6]).eval()
262 INPUT_MATRIX, col_slices=[3, 6], shuffle=True).eval()
[all …]
|
| D | wals_test.py | 46 INPUT_MATRIX = factorization_ops_test_utils.INPUT_MATRIX variable in WALSMatrixFactorizationTest
66 sp_inputs = self.np_array_to_sparse(self.INPUT_MATRIX)
184 return self.INPUT_MATRIX
289 np_matrix=self.INPUT_MATRIX[[1, 4], :],
306 np_matrix=self.INPUT_MATRIX[:, [5, 3, 1]],
388 WALSMatrixFactorizationTest.INPUT_MATRIX,
|
| D | factorization_ops_test_utils.py | 32 INPUT_MATRIX = np.array( variable
|