Searched refs:new_ids (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | generate_vocab_remapping_op.cc | 109 Tensor new_ids; in Compute() local
112 &new_ids)); in Compute()
113 auto new_ids_vec = new_ids.vec<int64>(); in Compute()
145 OP_REQUIRES_OK(context, new_vocab_table->Find(context, new_ids, &tokens, in Compute()
|
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| D | factorization_ops.py | 644 new_ids = array_ops.where(assignments < extras,
647 return assignments, new_ids
662 assignments, new_ids = sharding_func(indices)
665 sharded_ids = data_flow_ops.dynamic_partition(new_ids, assignments,
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | embedding_ops.py | 153 new_ids = flat_ids // np
184 new_ids = array_ops.where(p_assignments < extras,
195 gather_ids = data_flow_ops.dynamic_partition(new_ids, p_assignments, np)
|
| /external/tensorflow/tensorflow/contrib/image/python/ops/ |
| D | image_ops.py | 526 new_ids = control_flow_ops.cond(
529 array_ops.gather(new_ids, id_index), array_ops.shape(components))
|
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ops/ |
| D | sdca_ops.py | 465 new_ids = array_ops.where(p_assignments < extras,
474 gather_ids = data_flow_ops.dynamic_partition(new_ids,
492 math_ops.range(array_ops.shape(new_ids)[0]),
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | embedding_ops.py | 778 new_ids = flat_ids // np
807 new_ids = (is_in_first_extras_partitions * (flat_ids %
819 gather_ids = data_flow_ops.dynamic_partition(new_ids, p_assignments, np)
|
| /external/mesa3d/src/compiler/spirv/ |
| D | vtn_variables.c | 47 unsigned new_ids) in vtn_access_chain_extend() argument
52 chain = vtn_access_chain_create(b, old_len + new_ids); in vtn_access_chain_extend()
|