Searched defs:host_buffer (Results 1 – 10 of 10) 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().CopyRef(); in CopyRefDHTToRuntimeFallbackTensor() local
57 auto host_buffer = tfrt::HostBuffer::CreateFromExternal( in ConvertRuntimeFallbackTensorToDenseHostTensor() local
324 HostBuffer* host_buffer = dht.ReleaseBuffer().release(); in MoveDHTToTFTensor() local326 auto* host_buffer = reinterpret_cast<HostBuffer*>(arg); in MoveDHTToTFTensor() local1180 RCReference<HostBuffer> host_buffer = dht->buffer().CopyRef(); in ConvertTFRTTensorToTFTensorHandle() local1230 auto host_buffer = HostBuffer::CreateFromExternal( in ConvertTFTensorHandleToTFRTTensor() local
74 auto host_buffer = absl::make_unique<char[]>(buf.size()); in PrintBufferContents() local
391 static std::vector<T>* host_buffer = [] { in InitializeTypedBuffer() local
86 tfrt::RCReference<tfrt::HostBuffer> host_buffer = in ConvertKernelFallbackTensorToDenseHostTensor() local
106 for (auto host_buffer : arguments) { in RunMlirModuleWithHostBuffers() local
370 auto& host_buffer = compute_pipeline.pipeline->GetBuffers()[i]; in MapDeviceBufferToHostBuffer() local