Home
last modified time | relevance | path

Searched defs:host_buffer (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/util/
H A Dtensor_util.cc31 tensorflow::Tensor MoveHostBufferToTfTensor(RCReference<HostBuffer> host_buffer, in MoveHostBufferToTfTensor()
38 auto* host_buffer = reinterpret_cast<HostBuffer*>(arg); in MoveHostBufferToTfTensor() local
/aosp_15_r20/external/OpenCL-CTS/test_conformance/basic/
H A Dtest_simple_image_pitch.cpp57 char* host_buffer = (char*)malloc(buffer_bytes); in test_simple_read_image_pitch() local
120 char* host_buffer = (char*)malloc(buffer_bytes); in test_simple_write_image_pitch() local
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Druntime_fallback_tensor.cc174 RCReference<HostBuffer> host_buffer = dht.ReleaseBuffer(); in MoveDHTToRuntimeFallbackTensor() local
189 RCReference<HostBuffer> host_buffer = dht.buffer(); in CopyRefDHTToRuntimeFallbackTensor() local
H A Druntime_fallback_kernels.cc326 HostBuffer* host_buffer = dht.ReleaseBuffer().release(); in MoveDHTToTFTensor() local
328 auto* host_buffer = reinterpret_cast<HostBuffer*>(arg); in MoveDHTToTFTensor() local
1252 auto host_buffer = HostBuffer::CreateFromExternal( in ConvertTFTensorHandleToTFRTTensor() local
H A Dconversion_function.cc57 auto host_buffer = tfrt::HostBuffer::CreateFromExternal( in ConvertRuntimeFallbackTensorToDenseHostTensor() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dkernel_thunk.cc76 auto host_buffer = std::make_unique<char[]>(buf.size()); in PrintBufferContents() local
H A Dstream_executor_util.cc402 static std::vector<T>* host_buffer = [] { in InitializeTypedBuffer() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
H A Dmlir_gpu_test_base.cc110 for (auto host_buffer : arguments) { in RunMlirModuleWithHostBuffers() local
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/conversion/
H A Dconversion.cc109 tfrt::RCReference<tfrt::HostBuffer> host_buffer = in ConvertKernelFallbackTensorToDenseHostTensor() local
/aosp_15_r20/external/virtio-media/device/src/devices/
H A Dsimple_device.rs486 let host_buffer = session in qbuf() localVariable
H A Dvideo_decoder.rs1150 let host_buffer = buffers in qbuf() localVariable
H A Dv4l2_device_proxy.rs110 let mut host_buffer = guest_buffer.clone(); in guest_v4l2_buffer_to_host() localVariable
/aosp_15_r20/external/OpenCL-CTS/test_conformance/images/kernel_read_write/
H A Dtest_cl_ext_image_from_buffer.cpp870 std::vector<char> host_buffer(buffer_size); in image_from_buffer_read_check() local
/aosp_15_r20/external/deqp-deps/amber/src/dawn/
Dengine_dawn.cc370 auto& host_buffer = compute_pipeline.pipeline->GetBuffers()[i]; in MapDeviceBufferToHostBuffer() local