| /external/tensorflow/tensorflow/core/kernels/ |
| D | tensor_array.h | 188 PersistentTensor* value) { in WriteOrAggregate()
196 std::vector<PersistentTensor>* values) { in WriteOrAggregateMany()
224 PersistentTensor* value) { in Read()
231 std::vector<PersistentTensor>* values) { in ReadMany()
354 PersistentTensor* value) TF_EXCLUSIVE_LOCKS_REQUIRED(mu_);
358 PersistentTensor* value)
363 PersistentTensor* value) TF_EXCLUSIVE_LOCKS_REQUIRED(mu_);
422 PersistentTensor tensor;
444 PersistentTensor* value) { in LockedWriteOrAggregate()
523 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 | 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()
161 PersistentTensor priority_element; in TryEnqueueMany()
175 PersistentTensor element; in TryEnqueueMany()
|
| D | random_shuffle_queue_op.cc | 42 class RandomShuffleQueue : public TypedQueue<std::vector<PersistentTensor> > {
77 PersistentTensor* out_tensor);
148 queues_[i].push_back(PersistentTensor(tuple[i])); in TryEnqueue()
168 PersistentTensor* out_tensor) { in GetElementComponentFromBatch()
210 PersistentTensor element; in TryEnqueueMany()
350 PersistentTensor element; in TryDequeueMany()
|
| 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 | padding_fifo_queue.h | 79 PersistentTensor* out_tensor);
|
| 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 | lookup_table_op.cc | 366 empty_key_ = PersistentTensor(*empty_key_input); in MutableDenseHashTable()
378 deleted_key_ = PersistentTensor(*deleted_key_input); in MutableDenseHashTable()
517 key_buckets_ = PersistentTensor(keys); in ImportValues()
518 value_buckets_ = PersistentTensor(values); in ImportValues()
793 PersistentTensor key_buckets_ TF_GUARDED_BY(mu_);
794 PersistentTensor value_buckets_ TF_GUARDED_BY(mu_);
795 PersistentTensor empty_key_;
797 PersistentTensor deleted_key_;
|
| D | tensor_array_ops.cc | 441 PersistentTensor persistent_tensor(*tensor_value); in Compute()
523 PersistentTensor value; in Compute()
615 std::vector<PersistentTensor> values; in Compute()
860 std::vector<PersistentTensor> values; in Compute()
1120 std::vector<PersistentTensor> write_values; in Compute()
1125 PersistentTensor persistent_tensor; in Compute()
1305 std::vector<PersistentTensor> write_values; in Compute()
1310 PersistentTensor persistent_tensor; in Compute()
|
| D | queue_ops.cc | 94 PersistentTensor handle_;
|
| D | training_op_helpers.h | 44 PersistentTensor unused; in EnsureSparseVariableAccess()
203 PersistentTensor unused; in PrepareToUpdateVariable()
|
| D | assign_op.h | 113 PersistentTensor copy; in Compute()
|
| D | conditional_accumulator.h | 65 PersistentTensor accum_grad_persistent_;
|
| D | count_up_to_op.cc | 86 PersistentTensor unused; in Compute()
|
| 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/core/kernels/sparse/ |
| D | zeros_op.h | 70 PersistentTensor csr_row_ptr_pt; in operator()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | resource_op_kernel.h | 133 PersistentTensor handle_ TF_GUARDED_BY(mu_);
|
| D | op_kernel.h | 235 class PersistentTensor {
237 PersistentTensor() {} in PersistentTensor() function
238 explicit PersistentTensor(const Tensor& tensor) : tensor_(tensor) {} in PersistentTensor() function
300 PersistentTensor* out_persistent,
1033 PersistentTensor* out_persistent,
1036 PersistentTensor* out_persistent, in allocate_persistent()
|
| D | op_kernel.cc | 216 Tensor* PersistentTensor::AccessTensor(OpKernelConstruction* context) { in AccessTensor()
222 Tensor* PersistentTensor::AccessTensor(OpKernelContext* context) { in AccessTensor()
305 DataType type, const TensorShape& shape, PersistentTensor* out_persistent, in allocate_persistent()
312 *out_persistent = PersistentTensor(persistent); in allocate_persistent()
815 PersistentTensor* out_persistent, in allocate_persistent()
830 *out_persistent = PersistentTensor(persistent); in allocate_persistent()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_lru_cache.h | 174 std::vector<PersistentTensor> device_tensors_;
|