Searched refs:buffer_8 (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | gpu_transfer_manager.cc | 122 auto buffer_8 = se::DeviceMemory<uint8_t>(*buffer); in ReadDynamicShapes() local 124 stream->parent()->GetSubBuffer(&buffer_8, offset, metadata_size); in ReadDynamicShapes()
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | cpu_xfeed.cc | 302 auto buffer_8 = static_cast<int8_t*>(memory); in ReadDynamicShapesOnCpu() local 303 auto metadata_buffer = reinterpret_cast<int32_t*>(buffer_8 + offset); in ReadDynamicShapesOnCpu()
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | transfer_manager.cc | 228 auto buffer_8 = se::DeviceMemory<uint8_t>(*buffer); in ReadDynamicShapes() local 230 stream->parent()->GetSubBuffer(&buffer_8, offset, metadata_size); in ReadDynamicShapes()
|
/external/vulkan-validation-layers/tests/ |
D | vklayertests_command.cpp | 1297 VkBufferObj buffer_1024, buffer_64, buffer_16, buffer_8; in TEST_F() local 1301 buffer_8.init_as_src_and_dst(*m_device, 8, reqs); in TEST_F() 1349 …fer(m_commandBuffer->handle(), image.handle(), VK_IMAGE_LAYOUT_GENERAL, buffer_8.handle(), 1, ®… in TEST_F() 1352 …vkCmdCopyBufferToImage(m_commandBuffer->handle(), buffer_8.handle(), image.handle(), VK_IMAGE_LAYO… in TEST_F()
|