Searched defs:host_buffer (Results 1 – 11 of 11) sorted by relevance
31 tensorflow::Tensor MoveHostBufferToTfTensor(RCReference<HostBuffer> host_buffer, in MoveHostBufferToTfTensor()38 auto* host_buffer = reinterpret_cast<HostBuffer*>(arg); in MoveHostBufferToTfTensor() local
57 char* host_buffer = (char*)malloc(buffer_bytes); in test_simple_read_image_pitch() local120 char* host_buffer = (char*)malloc(buffer_bytes); in test_simple_write_image_pitch() local
174 RCReference<HostBuffer> host_buffer = dht.ReleaseBuffer(); in MoveDHTToRuntimeFallbackTensor() local189 RCReference<HostBuffer> host_buffer = dht.buffer(); in CopyRefDHTToRuntimeFallbackTensor() local
57 auto host_buffer = tfrt::HostBuffer::CreateFromExternal( in ConvertRuntimeFallbackTensorToDenseHostTensor() local
326 HostBuffer* host_buffer = dht.ReleaseBuffer().release(); in MoveDHTToTFTensor() local328 auto* host_buffer = reinterpret_cast<HostBuffer*>(arg); in MoveDHTToTFTensor() local1252 auto host_buffer = HostBuffer::CreateFromExternal( in ConvertTFTensorHandleToTFRTTensor() local
76 auto host_buffer = std::make_unique<char[]>(buf.size()); in PrintBufferContents() local
402 static std::vector<T>* host_buffer = [] { in InitializeTypedBuffer() local
110 for (auto host_buffer : arguments) { in RunMlirModuleWithHostBuffers() local
109 tfrt::RCReference<tfrt::HostBuffer> host_buffer = in ConvertKernelFallbackTensorToDenseHostTensor() local
863 std::vector<char> host_buffer(buffer_size); in image_from_buffer_read_check() local
370 auto& host_buffer = compute_pipeline.pipeline->GetBuffers()[i]; in MapDeviceBufferToHostBuffer() local