Home
last modified time | relevance | path

Searched defs:usage_records (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmemory_management.cc63 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()
Dmemory_management_test.cc108 std::vector<TensorUsageRecord<size_t>> usage_records{ in TEST() local
158 std::vector<TensorUsageRecord<size_t>> usage_records{ in TEST() local
213 std::vector<TensorUsageRecord<size_t>> usage_records{ in TEST() local
273 std::vector<TensorUsageRecord<BHWC>> usage_records{ in TEST() local
306 std::vector<TensorUsageRecord<uint2>> usage_records{ in TEST() local
344 std::vector<TensorUsageRecord<uint3>> usage_records{ in TEST() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/memory_management/
Dinternal_test.cc39 std::vector<TensorUsageRecord<size_t>> usage_records{ in TEST() local
58 std::vector<TensorUsageRecord<size_t>> usage_records{ in TEST() local
83 std::vector<TensorUsageRecord<size_t>> usage_records{ in TEST() local
Dequality_assignment.h36 const std::vector<TensorUsageRecord<TensorSizeT>>& usage_records, in EqualityAssignmentWithHash()
81 const std::vector<TensorUsageRecord<TensorSizeT>>& usage_records, in EqualityAssignment()
Dinternal.cc65 const std::vector<TensorUsageRecord<size_t>>& usage_records) { in CalculateTaskProfiles()
88 const std::vector<TensorUsageRecord<size_t>>& usage_records) { in CalculatePositionalMaximums()
Dnaive_assignment.h36 const std::vector<TensorUsageRecord<TensorSizeT>>& usage_records, in NaiveAssignment()
Dgreedy_by_size_assignment.cc69 const std::vector<TensorUsageRecord<size_t>>& usage_records, in GreedyBySizeAssignment()
161 const std::vector<TensorUsageRecord<size_t>>& usage_records, in GreedyBySizeDistPriorityAssignment()
Dmin_cost_flow_assignment.cc38 void Build(const std::vector<TensorUsageRecord<size_t>>& usage_records) { in Build()
219 const std::vector<TensorUsageRecord<size_t>>& usage_records, in MinCostFlowAssignment()
Dgreedy_in_order_assignment.h150 const std::vector<TensorUsageRecord<TensorSizeT>>& usage_records, in GreedyInOrderAssignmentMultidimensional()
Dgreedy_by_breadth_assignment.cc51 const std::vector<TensorUsageRecord<size_t>>& usage_records, in GreedyByBreadthAssignment()
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/
Druntime.cc454 CombinedUsageRecords* usage_records; member
461 absl::Status AddUsageRecord(CombinedUsageRecords* usage_records, in AddUsageRecord()
598 const CombinedUsageRecords& usage_records = it.second; in AssignInternalObjects() local
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dinference_context.cc88 std::map<ValueId, int2>* usage_records) { in AddUsage()
583 std::vector<TensorUsageRecord<DummyTensor>> usage_records; in AllocateMemoryForStrongShapes() local
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dinference_context.cc73 std::map<ValueId, int2>* usage_records) { in AddUsage()
553 std::vector<TensorUsageRecord<DummyTensor>> usage_records; in AllocateMemoryForStrongShapes() local