Searched refs:device_memory_usage (Results 1 – 4 of 4) sorted by relevance
96 se->device_memory_usage = DeviceMemoryUsage; in PopulateDefaultStreamExecutor()
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()
247 se_.device_memory_usage = [](const SP_Device* const device, in TEST_F()