Searched refs:getIBuffer (Results 1 – 4 of 4) sorted by relevance
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | TestMemoryDomain.cpp | 307 EXPECT_NE(m->getIBuffer(), nullptr); in TEST_P()
317 EXPECT_EQ(m->getIBuffer(), nullptr); in TEST_P()
345 EXPECT_NE(m->getIBuffer(), nullptr); in TEST_P()
355 EXPECT_EQ(m->getIBuffer(), nullptr); in TEST_P()
379 EXPECT_EQ(m->getIBuffer(), nullptr); in TEST_P()
402 EXPECT_EQ(m->getIBuffer(), nullptr); in TEST_P()
429 EXPECT_NE(m->getIBuffer(), nullptr); in TEST_P()
|
| /packages/modules/NeuralNetworks/runtime/ |
| D | Memory.cpp | 289 bool srcHasIBuffer = src.getIBuffer() != nullptr; in copyInternal()
290 bool dstHasIBuffer = dst.getIBuffer() != nullptr; in copyInternal()
292 return copyIBuffers(src.getIBuffer(), dst.getIBuffer(), srcMetadata); in copyInternal()
296 return copyMemoryToIBuffer(src.getMemory(), dst.getIBuffer(), srcMetadata.dimensions); in copyInternal()
298 return copyIBufferToMemory(src.getIBuffer(), dst.getMemory()); in copyInternal()
|
| D | Memory.h | 177 const SharedBuffer& getIBuffer() const { return kBuffer; } in getIBuffer() function
|
| D | ExecutionBuilder.cpp | 1623 if (memory->getIBuffer() != nullptr) { in computeOnCpuFallback()
1630 n = copyIBufferToMemory(memory->getIBuffer(), blobAhwb->getMemory()); in computeOnCpuFallback()
1653 if (memory->getIBuffer() != nullptr && isUsedAsOutput[i]) { in computeOnCpuFallback()
1654 n = copyMemoryToIBuffer(memories[i]->getMemory(), memory->getIBuffer(), {}); in computeOnCpuFallback()
|