HomeSort by relevance Sort by last modified time
    Searched refs:p_assignments (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/python/ops/
embedding_ops.py 164 # Create p_assignments and set new_ids depending on the strategy.
166 p_assignments = flat_ids % np
190 p_assignments = math_ops.maximum(
195 new_ids = array_ops.where(p_assignments < extras,
204 p_assignments = math_ops.cast(p_assignments, dtypes.int32)
206 gather_ids = data_flow_ops.dynamic_partition(new_ids, p_assignments, np)
209 p_assignments, np)
  /external/tensorflow/tensorflow/contrib/layers/python/layers/
embedding_ops.py     [all...]

Completed in 214 milliseconds