Searched refs:row_weights (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| D | factorization_ops.py | 197 row_weights=1, argument
256 assert (row_weights is None) == (col_weights is None)
263 row_weights, self._input_rows, self._num_row_shards, "row_weights")
277 row_weights, self._input_rows, self._num_row_shards, "row_weights")
309 def row_weights(self): member in WALSModel
773 row_weights=projection_weights)[0]
806 row_weights=projection_weights)[0]
812 row_weights=None): argument
921 if row_weights is None:
930 [check_ops.assert_less_equal(array_ops.rank(row_weights), 1)]):
[all …]
|
| D | factorization_ops_test.py | 118 row_weights = [[8., 18., 28., 38., 48.]]
135 row_weights=row_weights,
179 row_weights=self.row_wts,
348 row_weights=self.row_wts,
516 row_weights=None,
537 row_weights=0,
594 row_weights=None,
606 row_weights=[0] * 5,
687 row_weights=None,
717 row_weights=0,
[all …]
|
| D | factorization_ops_test_utils.py | 99 w0=1., row_weights=None, col_weights=None): argument
118 wr = (array_ops.expand_dims(row_weights, 1) if row_weights is not None
|
| D | wals.py | 174 row_weights=params["row_weights"],
469 row_weights=1, argument
550 row_weights,
|
| D | wals_test.py | 252 row_weights=self._row_weights,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | scale_and_translate_op.cc | 359 typename TTypes<float, 1>::ConstTensor row_weights( in Compute() local
369 row_weights, col_spans.span_size, col_starts, col_weights, image_data, in Compute()
464 typename TTypes<float, 1>::ConstTensor row_weights( in Compute() local
474 row_weights, col_spans.span_size, col_starts, col_weights, input_grad, in Compute()
558 typename TTypes<float, 1>::ConstTensor row_weights, in operator ()()
584 GatherRows(row_span_size, row_starts.data(), row_weights.data(), in operator ()()
|
| D | scale_and_translate_op.h | 63 typename TTypes<float, 1>::ConstTensor row_weights,
|