Searched refs:p_assignments (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ops/ |
| D | sdca_ops.py | 369 p_assignments, argument
373 p_assignments = p_assignments_by_var[v_num]
376 full_update, p_assignments, num_partitions)
460 p_assignments = math_ops.maximum(
465 new_ids = array_ops.where(p_assignments < extras,
471 p_assignments = math_ops.cast(p_assignments, dtypes.int32)
475 p_assignments,
479 p_assignments_by_var[v_num] = p_assignments
493 p_assignments, num_partitions)
554 gather_ids_by_var, w, u, p_assignments, num_partitions)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | embedding_ops.py | 152 p_assignments = flat_ids % np
179 p_assignments = math_ops.maximum(
184 new_ids = array_ops.where(p_assignments < extras,
193 p_assignments = math_ops.cast(p_assignments, dtypes.int32)
195 gather_ids = data_flow_ops.dynamic_partition(new_ids, p_assignments, np)
198 p_assignments, np)
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | embedding_ops.py | 777 p_assignments = flat_ids % np
801 p_assignments = math_ops.maximum(flat_ids // (ids_per_partition + 1), (
805 is_in_first_extras_partitions = math_ops.cast(p_assignments < extras,
817 p_assignments = math_ops.cast(p_assignments, dtypes.int32)
819 gather_ids = data_flow_ops.dynamic_partition(new_ids, p_assignments, np)
822 p_assignments, np)
|