Searched refs:SetDeviceMemorySize (Results 1 – 2 of 2) sorted by relevance
193 (*allocation)->SetDeviceMemorySize(); in CreateAndTransfer()216 (*allocation)->SetDeviceMemorySize(); in CreateUninitialized()229 (*allocation)->SetDeviceMemorySize(); in CreateFromBuffer()400 (*allocation)->SetDeviceMemorySize(); in MakeSubBuffer()404 void XRTTupleAllocation::SetDeviceMemorySize() { in SetDeviceMemorySize() function in tensorflow::XRTTupleAllocation545 allocation_tmp->SetDeviceMemorySize(); in MakeTuple()
267 void SetDeviceMemorySize();