Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/lookup_tables/
Dgeneric_table_op_kernels.cc72 std::vector<typename VariantSubType::value_type> keys_vec; in TensorInsertHelper() local
73 keys_vec.reserve(keys_flat.size()); in TensorInsertHelper()
75 keys_vec.emplace_back( in TensorInsertHelper()
78 return table->InsertOrAssign(keys_vec, values.flat<ValueType>()); in TensorInsertHelper()
141 std::vector<typename VariantSubType::value_type> keys_vec; in TensorLookupHelper() local
143 keys_vec.reserve(keys_size); in TensorLookupHelper()
145 keys_vec.emplace_back(*keys_flat(i).get<VariantSubType>()->get()); in TensorLookupHelper()
147 absl::Span<const typename VariantSubType::value_type> key_span(keys_vec); in TensorLookupHelper()
Dflat_hash_map_op_kernels.cc238 std::vector<typename absl::string_view> keys_vec; in Initialize() local
239 keys_vec.reserve(keys.size()); in Initialize()
241 keys_vec.push_back(*keys[i].get<absl::string_view>()); in Initialize()
243 return container->Initialize(keys_vec, values); in Initialize()
/external/tensorflow/tensorflow/core/kernels/
Dreader_ops.cc118 std::vector<string> keys_vec; in ComputeWithReader() local
119 keys_vec.reserve(num_records); in ComputeWithReader()
124 reader->ReadUpTo(num_records, queue, &keys_vec, &values_vec, context); in ComputeWithReader()
126 OP_REQUIRES(context, num_actually_read == keys_vec.size(), in ComputeWithReader()
145 keys_t(i) = std::move(keys_vec[i]); in ComputeWithReader()
Dbarrier_ops.cc311 auto keys_vec = keys.flat<string>(); in InsertOneLocked() local
316 element_ptr = gtl::FindOrNull(incomplete_, keys_vec(i)); in InsertOneLocked()
321 keys_vec(i), in InsertOneLocked()
328 &gtl::LookupOrInsert(&incomplete_, keys_vec(i), PersistentTuple()); in InsertOneLocked()
361 return errors::InvalidArgument("Key ", keys_vec(i), in InsertOneLocked()
395 ready_tuple[1].scalar<string>()() = keys_vec(i); // set the key in InsertOneLocked()
399 incomplete_.erase(incomplete_.find(keys_vec(i))); in InsertOneLocked()