Searched refs:ObjectsAssignment (Results 1 – 16 of 16) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | memory_management.h | 39 const ObjectsAssignment<size_t>& obj_assignment);
80 ObjectsAssignment<size_t>* assignment);
92 MemoryStrategy strategy, ObjectsAssignment<TensorSizeT>* assignment,
109 MemoryStrategy strategy, ObjectsAssignment<size_t>* assignment,
115 MemoryStrategy strategy, ObjectsAssignment<BHWC>* assignment,
121 MemoryStrategy strategy, ObjectsAssignment<uint2>* assignment,
127 MemoryStrategy strategy, ObjectsAssignment<uint3>* assignment,
|
| D | memory_management.cc | 38 size_t TotalSize(const ObjectsAssignment<size_t>& assignment) { in TotalSize()
46 const ObjectsAssignment<size_t>& obj_assignment) { in ObjectsToOffsets()
64 ObjectsAssignment<size_t>* assignment) { in BestGreedy()
67 ObjectsAssignment<size_t> assignment_by_breadth; in BestGreedy()
78 MemoryStrategy strategy, ObjectsAssignment<size_t>* assignment, in AssignObjectsToTensors()
106 MemoryStrategy strategy, ObjectsAssignment<BHWC>* assignment, in AssignObjectsToTensors()
123 MemoryStrategy strategy, ObjectsAssignment<uint2>* assignment, in AssignObjectsToTensors()
142 MemoryStrategy strategy, ObjectsAssignment<uint3>* assignment, in AssignObjectsToTensors()
165 ObjectsAssignment<size_t> objects_assignment; in AssignOffsetsToTensors()
|
| D | memory_management_test.cc | 35 ObjectsAssignment<size_t> objects_assignment; in TEST()
42 ObjectsAssignment<size_t> objects_assignment; in TEST()
56 ObjectsAssignment<size_t> objects_assignment; in TEST()
64 ObjectsAssignment<size_t> assignment; in TEST()
111 ObjectsAssignment<size_t> assignment; in TEST()
166 ObjectsAssignment<size_t> assignment; in TEST()
224 ObjectsAssignment<size_t> assignment; in TEST()
284 ObjectsAssignment<BHWC> assignment; in TEST()
317 ObjectsAssignment<uint2> assignment; in TEST()
355 ObjectsAssignment<uint3> assignment; in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/memory_management/ |
| D | min_cost_flow_assignment.cc | 131 void CalculateAssignment(ObjectsAssignment<size_t>* assignment) { in CalculateAssignment()
184 ObjectsAssignment<size_t>* assignment) { in AssignTensorsToNewSharedObject()
220 ObjectsAssignment<size_t>* assignment) { in MinCostFlowAssignment()
|
| D | equality_assignment.h | 37 ObjectsAssignment<TensorSizeT>* assignment) { in EqualityAssignmentWithHash()
82 ObjectsAssignment<TensorSizeT>* assignment) { in EqualityAssignment()
|
| D | greedy_by_breadth_assignment.h | 50 ObjectsAssignment<size_t>* assignment);
|
| D | min_cost_flow_assignment.h | 37 ObjectsAssignment<size_t>* assignment);
|
| D | greedy_by_size_assignment.h | 73 ObjectsAssignment<size_t>* assignment);
|
| D | naive_assignment.h | 37 ObjectsAssignment<TensorSizeT>* assignment) { in NaiveAssignment()
|
| D | greedy_in_order_assignment.h | 55 ObjectsAssignment<TensorSizeT>* assignment,
151 ObjectsAssignment<TensorSizeT>* assignment) { in GreedyInOrderAssignmentMultidimensional()
|
| D | types.h | 51 struct ObjectsAssignment { struct
|
| D | greedy_by_breadth_assignment.cc | 52 ObjectsAssignment<size_t>* assignment) { in GreedyByBreadthAssignment()
|
| D | greedy_by_size_assignment.cc | 162 ObjectsAssignment<size_t>* assignment) { in GreedyBySizeDistPriorityAssignment()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/ |
| D | runtime.cc | 489 const ObjectsAssignment<size_t>& assignment, in ApplyBuffersAssignment()
529 const ObjectsAssignment<ObjectSizeT>& assignment, in ApplyTexturesAssignment()
600 ObjectsAssignment<size_t> buffer_assignment; in AssignInternalObjects()
609 ObjectsAssignment<size_t> texture_1d_assignment; in AssignInternalObjects()
618 ObjectsAssignment<uint2> texture_2d_assignment; in AssignInternalObjects()
627 ObjectsAssignment<uint3> texture_3d_assignment; in AssignInternalObjects()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | inference_context.cc | 498 ObjectsAssignment<size_t> buffer_assignment; in AllocateMemoryForBuffers()
562 ObjectsAssignment<DummyTensor> assignment; in AllocateMemoryForStrongShapes()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | inference_context.cc | 545 ObjectsAssignment<size_t> buffer_assignment; in AllocateMemoryForBuffers()
592 ObjectsAssignment<DummyTensor> assignment; in AllocateMemoryForStrongShapes()
|