Home
last modified time | relevance | path

Searched refs:MemoryStrategy (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmemory_management.cc78 MemoryStrategy strategy, ObjectsAssignment<size_t>* assignment, in AssignObjectsToTensors()
81 case MemoryStrategy::NAIVE: in AssignObjectsToTensors()
83 case MemoryStrategy::EQUALITY: in AssignObjectsToTensors()
85 case MemoryStrategy::GREEDY_IN_ORDER: in AssignObjectsToTensors()
88 case MemoryStrategy::GREEDY_BY_BREADTH: in AssignObjectsToTensors()
90 case MemoryStrategy::GREEDY_BY_SIZE: in AssignObjectsToTensors()
92 case MemoryStrategy::GREEDY_BEST: in AssignObjectsToTensors()
94 case MemoryStrategy::MINCOSTFLOW: in AssignObjectsToTensors()
106 MemoryStrategy strategy, ObjectsAssignment<BHWC>* assignment, in AssignObjectsToTensors()
109 case MemoryStrategy::NAIVE: in AssignObjectsToTensors()
[all …]
Dmemory_management_test.cc66 AssignObjectsToTensors({}, MemoryStrategy::NAIVE, &assignment).ok()); in TEST()
71 AssignObjectsToTensors({}, MemoryStrategy::EQUALITY, &assignment).ok()); in TEST()
76 AssignObjectsToTensors({}, MemoryStrategy::GREEDY_IN_ORDER, &assignment) in TEST()
82 AssignObjectsToTensors({}, MemoryStrategy::MINCOSTFLOW, &assignment) in TEST()
88 AssignObjectsToTensors({}, MemoryStrategy::GREEDY_BY_BREADTH, &assignment) in TEST()
94 AssignObjectsToTensors({}, MemoryStrategy::GREEDY_BY_SIZE, &assignment) in TEST()
100 ASSERT_TRUE(AssignOffsetsToTensors({}, MemoryStrategy::GREEDY_BY_SIZE, in TEST()
113 AssignObjectsToTensors(usage_records, MemoryStrategy::NAIVE, &assignment) in TEST()
118 ASSERT_TRUE(AssignObjectsToTensors(usage_records, MemoryStrategy::EQUALITY, in TEST()
125 usage_records, MemoryStrategy::GREEDY_IN_ORDER, &assignment) in TEST()
[all …]
Dmemory_management.h41 enum class MemoryStrategy { enum
92 MemoryStrategy strategy, ObjectsAssignment<TensorSizeT>* assignment,
95 case MemoryStrategy::NAIVE:
97 case MemoryStrategy::EQUALITY:
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,
134 const MemoryStrategy& strategy, OffsetsAssignment* assignment,
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/
Druntime.cc602 MemoryStrategy::GREEDY_BEST, in AssignInternalObjects()
611 MemoryStrategy::GREEDY_BEST, in AssignInternalObjects()
620 MemoryStrategy::GREEDY_IN_ORDER, in AssignInternalObjects()
629 MemoryStrategy::GREEDY_IN_ORDER, in AssignInternalObjects()
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dinference_context.cc500 buffer_usage_records, MemoryStrategy::GREEDY_BEST, &buffer_assignment)); in AllocateMemoryForBuffers()
564 usage_records, MemoryStrategy::EQUALITY, &assignment)); in AllocateMemoryForStrongShapes()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dinference_context.cc547 buffer_usage_records, MemoryStrategy::GREEDY_BEST, &buffer_assignment)); in AllocateMemoryForBuffers()
594 usage_records, MemoryStrategy::EQUALITY, &assignment)); in AllocateMemoryForStrongShapes()