| /external/tensorflow/tensorflow/core/kernels/ |
| D | tensor_array.h | 190 PersistentTensor* value) { in WriteOrAggregate()
198 std::vector<PersistentTensor>* values) { in WriteOrAggregateMany()
226 PersistentTensor* value) { in Read()
233 std::vector<PersistentTensor>* values) { in ReadMany()
355 PersistentTensor* value) EXCLUSIVE_LOCKS_REQUIRED(mu_);
359 PersistentTensor* value)
364 PersistentTensor* value) EXCLUSIVE_LOCKS_REQUIRED(mu_);
423 PersistentTensor tensor;
445 PersistentTensor* value) { in LockedWriteOrAggregate()
524 PersistentTensor local_tensor; in LockedWriteOrAggregate()
[all …]
|
| D | typed_queue.h | 83 int64 SizeOf(const std::deque<PersistentTensor>& sq) { in SizeOf()
91 int64 SizeOf(const std::vector<PersistentTensor>& sq) { in SizeOf()
98 using TensorPair = std::pair<int64, PersistentTensor>;
|
| D | fifo_queue.h | 34 class FIFOQueue : public TypedQueue<std::deque<PersistentTensor> > {
67 PersistentTensor* out_element);
|
| D | priority_queue.h | 34 using PriorityTensorPair = std::pair<int64, PersistentTensor>;
86 PersistentTensor* out_element);
|
| D | random_shuffle_queue_op.cc | 42 class RandomShuffleQueue : public TypedQueue<std::vector<PersistentTensor> > {
77 PersistentTensor* out_tensor);
147 queues_[i].push_back(PersistentTensor(tuple[i])); in TryEnqueue()
167 PersistentTensor* out_tensor) { in GetElementComponentFromBatch()
209 PersistentTensor element; in TryEnqueueMany()
349 PersistentTensor element; in TryDequeueMany()
|
| D | priority_queue.cc | 66 PersistentTensor persistent_tensor = gtl::ConsumeTop(&queues_[i]).second; in DequeueLocked()
99 queues_[i].emplace(priority, PersistentTensor(tuple[i])); in TryEnqueue()
119 OpKernelContext* ctx, PersistentTensor* out_tensor) { in GetElementComponentFromBatch()
160 PersistentTensor priority_element; in TryEnqueueMany()
174 PersistentTensor element; in TryEnqueueMany()
|
| D | barrier_ops.cc | 342 PersistentTensor index_persistent_tensor; in InsertOneLocked()
356 element.push_back(PersistentTensor(uninitialized)); in InsertOneLocked()
359 const PersistentTensor& component = element[1 + component_index]; in InsertOneLocked()
368 PersistentTensor next_element; in InsertOneLocked()
389 PersistentTensor key; in InsertOneLocked()
426 typedef std::vector<PersistentTensor> PersistentTuple;
|
| D | fifo_queue.cc | 70 queues_[i].push_back(PersistentTensor(tuple[i])); in TryEnqueue()
91 PersistentTensor* out_tensor) { in GetElementComponentFromBatch()
132 PersistentTensor element; in TryEnqueueMany()
272 PersistentTensor element; in TryDequeueMany()
|
| D | sparse_conditional_accumulator.h | 60 accum_val_persistent_ = new PersistentTensor(); in SparseConditionalAccumulator()
75 PersistentTensor* accum_val_persistent_ = nullptr;
256 PersistentTensor* tensor_sum_persistent = new PersistentTensor(); in AddToAccumGradFunction()
|
| D | lookup_table_op.cc | 344 empty_key_ = PersistentTensor(*empty_key_input); in MutableDenseHashTable()
356 deleted_key_ = PersistentTensor(*deleted_key_input); in MutableDenseHashTable()
495 key_buckets_ = PersistentTensor(keys); in ImportValues()
496 value_buckets_ = PersistentTensor(values); in ImportValues()
771 PersistentTensor key_buckets_ GUARDED_BY(mu_);
772 PersistentTensor value_buckets_ GUARDED_BY(mu_);
773 PersistentTensor empty_key_;
775 PersistentTensor deleted_key_;
|
| D | sparse_tensors_map_ops.cc | 49 PersistentTensor indices;
50 PersistentTensor values;
56 PersistentTensor persistent_ix; in AddSparseTensor()
62 PersistentTensor persistent_values; in AddSparseTensor()
|
| D | tensor_array_ops.cc | 444 PersistentTensor persistent_tensor(*tensor_value); in Compute()
527 PersistentTensor value; in Compute()
620 std::vector<PersistentTensor> values; in Compute()
866 std::vector<PersistentTensor> values; in Compute()
1127 std::vector<PersistentTensor> write_values; in Compute()
1132 PersistentTensor persistent_tensor; in Compute()
1313 std::vector<PersistentTensor> write_values; in Compute()
1318 PersistentTensor persistent_tensor; in Compute()
|
| D | padding_fifo_queue.h | 79 PersistentTensor* out_tensor);
|
| D | training_op_helpers.h | 43 PersistentTensor unused; in EnsureSparseVariableAccess()
204 PersistentTensor unused; in PrepareToUpdateVariable()
|
| D | queue_ops.cc | 76 PersistentTensor handle_;
|
| D | assign_op.h | 113 PersistentTensor copy; in Compute()
|
| D | count_up_to_op.cc | 87 PersistentTensor unused; in Compute()
|
| D | conditional_accumulator.h | 65 PersistentTensor accum_grad_persistent_;
|
| D | dense_update_functor.cc | 86 PersistentTensor tmp; \
|
| D | padding_fifo_queue.cc | 60 PersistentTensor* out_tensor) { in GetElementComponent()
110 PersistentTensor element; in TryDequeueMany()
|
| /external/tensorflow/tensorflow/contrib/framework/kernels/ |
| D | zero_initializer_op.cc | 49 PersistentTensor out_persistent; in Compute()
103 PersistentTensor unused; in Compute()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_resources.h | 57 std::vector<PersistentTensor> device_tensors_;
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 242 class PersistentTensor {
244 PersistentTensor() {} in PersistentTensor() function
245 explicit PersistentTensor(const Tensor& tensor) : tensor_(tensor) {} in PersistentTensor() function
306 PersistentTensor* out_persistent,
989 PersistentTensor* out_persistent,
992 PersistentTensor* out_persistent, in allocate_persistent()
1255 friend class PersistentTensor; variable
|
| D | resource_op_kernel.h | 132 PersistentTensor handle_ GUARDED_BY(mu_);
|
| D | op_kernel.cc | 188 Tensor* PersistentTensor::AccessTensor(OpKernelConstruction* context) { in AccessTensor()
194 Tensor* PersistentTensor::AccessTensor(OpKernelContext* context) { in AccessTensor()
255 DataType type, const TensorShape& shape, PersistentTensor* out_persistent, in allocate_persistent()
264 *out_persistent = PersistentTensor(persistent); in allocate_persistent()
760 PersistentTensor* out_persistent, in allocate_persistent()
766 *out_persistent = PersistentTensor(persistent); in allocate_persistent()
|