Searched refs:device_memory_usage (Results 1 – 3 of 3) sorted by relevance
61 TF_Bool device_memory_usage(const SP_Device* const device, int64_t* const free, in device_memory_usage() function124 se->device_memory_usage = device_memory_usage; in PopulateDefaultStreamExecutor()425 se_.device_memory_usage = [](const SP_Device* const device, in TEST_F()
278 TF_Bool (*device_memory_usage)(const SP_Device* device, int64_t* free, member
157 VALIDATE_MEMBER(SP_StreamExecutor, se, device_memory_usage); in ValidateSPStreamExecutor()598 return stream_executor_->device_memory_usage( in DeviceMemoryUsage()