Searched refs:row_ids (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| D | wals_test.py | 72 def remap_sparse_tensor_rows(self, sp_x, row_ids, shape): argument
76 new_row_ids = array_ops.gather(row_ids, old_row_ids)
129 row_ids = row_batch[0]
132 row_batch[1], row_ids, shape=[num_rows, num_cols])
144 row_ids = math_ops.range(num_rows, dtype=dtypes.int64)
149 [row_ids, sp_mat],
|
| D | factorization_ops.py | 873 row_ids, col_ids = array_ops.split(
875 update_row_indices, all_row_ids = array_ops.unique(row_ids[:, 0])
878 row_ids = array_ops.expand_dims(math_ops.cast(all_row_ids, dtypes.int64), 1)
897 new_sp_indices = array_ops.concat([row_ids, col_ids], 1)
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | utils.py | 33 row_ids = sp_tensor.indices[:, 0]
38 seq_length = math_ops.segment_max(column_ids, segment_ids=row_ids)
|