Searched refs:keys_vec (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reader_ops.cc | 121 std::vector<tstring> keys_vec; in ComputeWithReader() local
122 keys_vec.reserve(num_records); in ComputeWithReader()
127 reader->ReadUpTo(num_records, queue, &keys_vec, &values_vec, context); in ComputeWithReader()
129 OP_REQUIRES(context, num_actually_read == keys_vec.size(), in ComputeWithReader()
148 keys_t(i) = std::move(keys_vec[i]); in ComputeWithReader()
|
| D | barrier_ops.cc | 311 auto keys_vec = keys.flat<tstring>(); in InsertOneLocked() local
316 element_ptr = gtl::FindOrNull(incomplete_, keys_vec(i)); in InsertOneLocked()
321 keys_vec(i), in InsertOneLocked()
328 >l::LookupOrInsert(&incomplete_, keys_vec(i), PersistentTuple()); in InsertOneLocked()
361 return errors::InvalidArgument("Key ", keys_vec(i), in InsertOneLocked()
395 ready_tuple[1].scalar<tstring>()() = keys_vec(i); // set the key in InsertOneLocked()
399 incomplete_.erase(incomplete_.find(keys_vec(i))); in InsertOneLocked()
|