Searched refs:col_weights (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| D | factorization_ops_test.py | 119 col_weights = [[90., 91., 92., 93., 94.]]
136 col_weights=col_weights,
180 col_weights=self.col_wts,
349 col_weights=self.col_wts,
517 col_weights=None,
538 col_weights=0,
595 col_weights=None,
607 col_weights=[0] * 7,
688 col_weights=None,
718 col_weights=[0] * cols,
[all …]
|
| D | factorization_ops_test_utils.py | 99 w0=1., row_weights=None, col_weights=None): argument
120 wc = (array_ops.expand_dims(col_weights, 0) if col_weights is not None
|
| D | factorization_ops.py | 198 col_weights=1, argument
256 assert (row_weights is None) == (col_weights is None)
266 col_weights, self._input_cols, self._num_col_shards, "col_weights")
279 col_weights, self._input_cols, self._num_col_shards, "col_weights")
314 def col_weights(self): member in WALSModel
936 col_weights = embedding_ops.embedding_lookup(
941 col_weights,
984 col_weights, 0))
|
| D | wals.py | 175 col_weights=params["col_weights"],
470 col_weights=1, argument
552 col_weights,
|
| D | wals_test.py | 253 col_weights=self._col_weights,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | scale_and_translate_op.cc | 364 typename TTypes<float, 1>::ConstTensor col_weights( in Compute() local
369 row_weights, col_spans.span_size, col_starts, col_weights, image_data, in Compute()
469 typename TTypes<float, 1>::ConstTensor col_weights( in Compute() local
474 row_weights, col_spans.span_size, col_starts, col_weights, input_grad, in Compute()
561 typename TTypes<float, 1>::ConstTensor col_weights, in operator ()()
587 GatherColumns(col_span_size, col_starts.data(), col_weights.data(), in operator ()()
|
| D | scale_and_translate_op.h | 66 typename TTypes<float, 1>::ConstTensor col_weights,
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | feature_column_test.py | 1162 _, col_weights, _ = (
1170 for weight in col_weights.values():
1182 saved_col_weights = col_weights[crossed_col][0].eval()
1197 _, col_weights, _ = (
1202 col_weights_from_ckpt = col_weights[crossed_col_initialized][0]
|