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
167 RETURN_IF_ERROR(AssignObjectsToTensors( in AssignOffsetsToTensors()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/ |
| D | runtime.cc | 604 RETURN_IF_ERROR(AssignObjectsToTensors(usage_records.buffers, in AssignInternalObjects()
613 RETURN_IF_ERROR(AssignObjectsToTensors(usage_records.textures_1d, in AssignInternalObjects()
622 RETURN_IF_ERROR(AssignObjectsToTensors(usage_records.textures_2d, in AssignInternalObjects()
631 RETURN_IF_ERROR(AssignObjectsToTensors(usage_records.textures_3d, in AssignInternalObjects()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | inference_context.cc | 516 RETURN_IF_ERROR(AssignObjectsToTensors( in AllocateMemoryForBuffers()
626 RETURN_IF_ERROR(AssignObjectsToTensors( in AllocateMemoryForStrongShapes()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | inference_context.cc | 197 RETURN_IF_ERROR(AssignObjectsToTensors( in GetBufferAsignment()
635 RETURN_IF_ERROR(AssignObjectsToTensors( in AllocateStrongShapesTensors()
|