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,
|