Searched refs:getIBuffer (Results 1 – 4 of 4) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/runtime/test/ |
| H A D | TestMemoryDomain.cpp | 304 EXPECT_NE(m->getIBuffer(), nullptr); in TEST_P()
314 EXPECT_EQ(m->getIBuffer(), nullptr); in TEST_P()
342 EXPECT_NE(m->getIBuffer(), nullptr); in TEST_P()
352 EXPECT_EQ(m->getIBuffer(), nullptr); in TEST_P()
376 EXPECT_EQ(m->getIBuffer(), nullptr); in TEST_P()
399 EXPECT_EQ(m->getIBuffer(), nullptr); in TEST_P()
426 EXPECT_NE(m->getIBuffer(), nullptr); in TEST_P()
|
| /aosp12/packages/modules/NeuralNetworks/runtime/ |
| H A D | Memory.cpp | 281 bool srcHasIBuffer = src.getIBuffer() != nullptr; in copyInternal()
282 bool dstHasIBuffer = dst.getIBuffer() != nullptr; in copyInternal()
284 return copyIBuffers(src.getIBuffer(), dst.getIBuffer(), srcMetadata); in copyInternal()
288 return copyMemoryToIBuffer(src.getMemory(), dst.getIBuffer(), srcMetadata.dimensions); in copyInternal()
290 return copyIBufferToMemory(src.getIBuffer(), dst.getMemory()); in copyInternal()
|
| H A D | Memory.h | 177 const SharedBuffer& getIBuffer() const { return kBuffer; } in getIBuffer() function
|
| H A D | ExecutionBuilder.cpp | 1574 if (memory->getIBuffer() != nullptr) { in computeOnCpuFallback()
1581 n = copyIBufferToMemory(memory->getIBuffer(), blobAhwb->getMemory()); in computeOnCpuFallback()
1604 if (memory->getIBuffer() != nullptr && isUsedAsOutput[i]) { in computeOnCpuFallback()
1605 n = copyMemoryToIBuffer(memories[i]->getMemory(), memory->getIBuffer(), {}); in computeOnCpuFallback()
|