Searched refs:TFE_TensorHandleDeviceMemorySize (Results 1 – 3 of 3) sorted by relevance
384 TF_CAPI_EXPORT extern size_t TFE_TensorHandleDeviceMemorySize(TFE_TensorHandle*,
389 size_t size = TFE_TensorHandleDeviceMemorySize(copy, status); in TEST()
614 size_t TFE_TensorHandleDeviceMemorySize(TFE_TensorHandle* h, in TFE_TensorHandleDeviceMemorySize() function