| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | memory_management.h | 79 const std::vector<TensorUsageRecord<size_t>>& usage_records,
91 const std::vector<TensorUsageRecord<TensorSizeT>>& usage_records,
108 const std::vector<TensorUsageRecord<size_t>>& usage_records,
114 const std::vector<TensorUsageRecord<BHWC>>& usage_records,
120 const std::vector<TensorUsageRecord<uint2>>& usage_records,
126 const std::vector<TensorUsageRecord<uint3>>& usage_records,
133 const std::vector<TensorUsageRecord<size_t>>& usage_records,
|
| D | memory_management.cc | 63 const std::vector<TensorUsageRecord<size_t>>& usage_records, in BestGreedy()
77 const std::vector<TensorUsageRecord<size_t>>& usage_records, in AssignObjectsToTensors()
105 const std::vector<TensorUsageRecord<BHWC>>& usage_records, in AssignObjectsToTensors()
122 const std::vector<TensorUsageRecord<uint2>>& usage_records, in AssignObjectsToTensors()
141 const std::vector<TensorUsageRecord<uint3>>& usage_records, in AssignObjectsToTensors()
159 const std::vector<TensorUsageRecord<size_t>>& usage_records, in AssignOffsetsToTensors()
|
| D | memory_management_test.cc | 108 std::vector<TensorUsageRecord<size_t>> usage_records{ in TEST()
158 std::vector<TensorUsageRecord<size_t>> usage_records{ in TEST()
213 std::vector<TensorUsageRecord<size_t>> usage_records{ in TEST()
273 std::vector<TensorUsageRecord<BHWC>> usage_records{ in TEST()
306 std::vector<TensorUsageRecord<uint2>> usage_records{ in TEST()
344 std::vector<TensorUsageRecord<uint3>> usage_records{ in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/memory_management/ |
| D | internal.h | 37 const TensorUsageRecord<TensorSizeT>* usage_record;
40 TensorUsageWithIndex(const TensorUsageRecord<TensorSizeT>* usage_record, in TensorUsageWithIndex()
100 const std::vector<TensorUsageRecord<size_t>>& usage_records);
104 const std::vector<TensorUsageRecord<size_t>>& usage_records);
|
| D | types.h | 35 struct TensorUsageRecord { struct
40 TensorUsageRecord(TensorSizeT size, TaskId first, TaskId last) in TensorUsageRecord() argument
44 bool operator<(const TensorUsageRecord<TensorSizeT>& other) const {
|
| D | greedy_by_size_assignment.h | 44 const std::vector<TensorUsageRecord<size_t>>& usage_records,
72 const std::vector<TensorUsageRecord<size_t>>& usage_records,
|
| D | internal_test.cc | 39 std::vector<TensorUsageRecord<size_t>> usage_records{ in TEST()
58 std::vector<TensorUsageRecord<size_t>> usage_records{ in TEST()
83 std::vector<TensorUsageRecord<size_t>> usage_records{ in TEST()
|
| D | min_cost_flow_assignment.cc | 38 void Build(const std::vector<TensorUsageRecord<size_t>>& usage_records) { in Build()
204 const std::vector<TensorUsageRecord<size_t>>* usage_records_;
219 const std::vector<TensorUsageRecord<size_t>>& usage_records, in MinCostFlowAssignment()
|
| D | greedy_by_size_assignment.cc | 69 const std::vector<TensorUsageRecord<size_t>>& usage_records, in GreedyBySizeAssignment()
86 const TensorUsageRecord<size_t>* rec = rec_with_idx.usage_record; in GreedyBySizeAssignment()
161 const std::vector<TensorUsageRecord<size_t>>& usage_records, in GreedyBySizeDistPriorityAssignment()
|
| D | greedy_by_breadth_assignment.cc | 33 using SharedObjectSchedule = std::set<TensorUsageRecord<size_t>>;
51 const std::vector<TensorUsageRecord<size_t>>& usage_records, in GreedyByBreadthAssignment()
|
| D | equality_assignment.h | 36 const std::vector<TensorUsageRecord<TensorSizeT>>& usage_records, in EqualityAssignmentWithHash()
81 const std::vector<TensorUsageRecord<TensorSizeT>>& usage_records, in EqualityAssignment()
|
| D | internal.cc | 65 const std::vector<TensorUsageRecord<size_t>>& usage_records) { in CalculateTaskProfiles()
88 const std::vector<TensorUsageRecord<size_t>>& usage_records) { in CalculatePositionalMaximums()
|
| D | greedy_by_breadth_assignment.h | 49 const std::vector<TensorUsageRecord<size_t>>& usage_records,
|
| D | min_cost_flow_assignment.h | 36 const std::vector<TensorUsageRecord<size_t>>& usage_records,
|
| D | naive_assignment.h | 36 const std::vector<TensorUsageRecord<TensorSizeT>>& usage_records, in NaiveAssignment()
|
| D | greedy_in_order_assignment.h | 54 const std::vector<TensorUsageRecord<TensorSizeT>>& usage_records,
150 const std::vector<TensorUsageRecord<TensorSizeT>>& usage_records, in GreedyInOrderAssignmentMultidimensional()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/ |
| D | runtime.cc | 403 std::vector<TensorUsageRecord<size_t>> buffers;
404 std::vector<TensorUsageRecord<size_t>> textures_1d;
405 std::vector<TensorUsageRecord<uint2>> textures_2d;
406 std::vector<TensorUsageRecord<uint3>> textures_3d;
411 void UpdateUsageRecord(TensorUsageRecord<TensorSizeT>* usage_rec, in UpdateUsageRecord()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | inference_context.cc | 486 std::vector<TensorUsageRecord<size_t>> buffer_usage_records; in AllocateMemoryForBuffers()
553 std::vector<TensorUsageRecord<DummyTensor>> usage_records; in AllocateMemoryForStrongShapes()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | inference_context.cc | 529 std::vector<TensorUsageRecord<size_t>> buffer_usage_records; in AllocateMemoryForBuffers()
583 std::vector<TensorUsageRecord<DummyTensor>> usage_records; in AllocateMemoryForStrongShapes()
|