Searched refs:factor_dim (Results 1 - 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/factorization/kernels/ |
| H A D | wals_solver_ops.cc | 100 const int64 factor_dim = factors.dim_size(1); variable
110 factor_dim, factors_size);
117 0, TensorShape({block_size, factor_dim, factor_dim}),
123 1, TensorShape({block_size, factor_dim}),
126 factor_dim, block_size);
221 input_index * factor_dim * factor_dim,
222 factor_dim, factor_dim);
[all...] |
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| H A D | factorization_ops_test.py | 169 factor_dim = 3
173 factor_dim,
338 factor_dim = 3
342 factor_dim,
|
Completed in 115 milliseconds