Searched refs:memoryBuilder (Results 1 – 2 of 2) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/tools/test_generator/test_harness/ |
| H A D | TestUtils.cpp | 44 ConstantMemoryBuilder* memoryBuilder) { in createOperand() argument
46 CHECK(memoryBuilder != nullptr); in createOperand()
62 location = memoryBuilder->append(operand.data.get<void>(), operand.data.size()); in createOperand()
89 ConstantMemoryBuilder* memoryBuilder) { in createSubgraph() argument
95 [operandValues, memoryBuilder](const TestOperand& operand) { in createSubgraph()
96 return createOperand(operand, operandValues, memoryBuilder); in createSubgraph()
119 ConstantMemoryBuilder memoryBuilder(0); in createModel() local
126 [&operandValues, &memoryBuilder](const TestSubgraph& testSubgraph) { in createModel()
127 return createSubgraph(testSubgraph, &operandValues, &memoryBuilder); in createModel()
132 if (!memoryBuilder.empty()) { in createModel()
[all …]
|
| /aosp12/hardware/interfaces/neuralnetworks/utils/common/src/ |
| H A D | CommonUtils.cpp | 70 CHECK(memoryBuilder != nullptr); in copyPointersToSharedMemory()
80 operand->location = memoryBuilder->append(data, operand->location.length); in copyPointersToSharedMemory()
84 nn::ConstantMemoryBuilder* memoryBuilder) { in copyPointersToSharedMemory() argument
87 [memoryBuilder](auto& operand) { in copyPointersToSharedMemory()
88 copyPointersToSharedMemory(&operand, memoryBuilder); in copyPointersToSharedMemory()
187 nn::ConstantMemoryBuilder memoryBuilder(modelInShared.pools.size()); in flushDataFromPointerToShared() local
188 copyPointersToSharedMemory(&modelInShared.main, &memoryBuilder); in flushDataFromPointerToShared()
190 [&memoryBuilder](auto& subgraph) { in flushDataFromPointerToShared()
191 copyPointersToSharedMemory(&subgraph, &memoryBuilder); in flushDataFromPointerToShared()
194 if (!memoryBuilder.empty()) { in flushDataFromPointerToShared()
[all …]
|