Searched refs:factor_dim (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/factorization/kernels/ |
| D | wals_solver_ops.cc | 122 const int64 factor_dim = factors.dim_size(1); in Compute() local
133 factor_dim, factors_size); in Compute()
140 0, TensorShape({block_size, factor_dim, factor_dim}), in Compute()
146 1, TensorShape({block_size, factor_dim}), in Compute()
149 factor_dim, block_size); in Compute()
250 input_index * factor_dim * factor_dim, in Compute()
251 factor_dim, factor_dim); in Compute()
|
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| D | factorization_ops_test.py | 169 factor_dim = 3
173 factor_dim,
338 factor_dim = 3
342 factor_dim,
|