Home
last modified time | relevance | path

Searched refs:sharding_func (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/contrib/factorization/python/ops/
Dfactorization_ops.py652 def scatter_update(cls, factor, indices, values, sharding_func, name=None): argument
662 assignments, new_ids = sharding_func(indices)
853 sharding_func = WALSModel._get_sharding_func(self._input_rows,
863 sharding_func = WALSModel._get_sharding_func(self._input_cols,
960 sharding_func,