Searched refs:AssignObjectsToTensors (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | memory_management_test.cc | 66 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()
113 AssignObjectsToTensors(usage_records, MemoryStrategy::NAIVE, &assignment) in TEST()
118 ASSERT_TRUE(AssignObjectsToTensors(usage_records, MemoryStrategy::EQUALITY, in TEST()
124 ASSERT_TRUE(AssignObjectsToTensors( in TEST()
130 ASSERT_TRUE(AssignObjectsToTensors(usage_records, MemoryStrategy::MINCOSTFLOW, in TEST()
[all …]
|
| D | memory_management.h | 90 absl::Status AssignObjectsToTensors(
107 absl::Status AssignObjectsToTensors(
113 absl::Status AssignObjectsToTensors(
119 absl::Status AssignObjectsToTensors(
125 absl::Status AssignObjectsToTensors(
|
| D | memory_management.cc | 76 absl::Status AssignObjectsToTensors( in AssignObjectsToTensors() function
104 absl::Status AssignObjectsToTensors( in AssignObjectsToTensors() function
121 absl::Status AssignObjectsToTensors( in AssignObjectsToTensors() function
140 absl::Status AssignObjectsToTensors( in AssignObjectsToTensors() function
166 RETURN_IF_ERROR(AssignObjectsToTensors( in AssignOffsetsToTensors()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/ |
| D | runtime.cc | 601 RETURN_IF_ERROR(AssignObjectsToTensors(usage_records.buffers, in AssignInternalObjects()
610 RETURN_IF_ERROR(AssignObjectsToTensors(usage_records.textures_1d, in AssignInternalObjects()
619 RETURN_IF_ERROR(AssignObjectsToTensors(usage_records.textures_2d, in AssignInternalObjects()
628 RETURN_IF_ERROR(AssignObjectsToTensors(usage_records.textures_3d, in AssignInternalObjects()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | inference_context.cc | 499 RETURN_IF_ERROR(AssignObjectsToTensors( in AllocateMemoryForBuffers()
563 RETURN_IF_ERROR(AssignObjectsToTensors( in AllocateMemoryForStrongShapes()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | inference_context.cc | 546 RETURN_IF_ERROR(AssignObjectsToTensors( in AllocateMemoryForBuffers()
593 RETURN_IF_ERROR(AssignObjectsToTensors( in AllocateMemoryForStrongShapes()
|