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