Home
last modified time | relevance | path

Searched refs:ExecutorForPlatformDeviceId (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/device/
Ddevice_id_utils.h34 static se::port::StatusOr<se::StreamExecutor*> ExecutorForPlatformDeviceId( in ExecutorForPlatformDeviceId() function
44 return ExecutorForPlatformDeviceId(device_manager, platform_device_id); in ExecutorForTfDeviceId()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_virtual_mem_allocator_test.cc40 DeviceIdUtil::ExecutorForPlatformDeviceId(GPUMachineManager(), gpu_id) in CreateAllocator()
53 DeviceIdUtil::ExecutorForPlatformDeviceId(GPUMachineManager(), gpu_id) in TEST()
Dgpu_debug_allocator.cc81 stream_exec_ = DeviceIdUtil::ExecutorForPlatformDeviceId(GPUMachineManager(), in GPUDebugAllocator()
160 stream_exec_ = DeviceIdUtil::ExecutorForPlatformDeviceId(GPUMachineManager(), in GPUNanResetAllocator()
Dgpu_cudamalloc_allocator.cc32 stream_exec_ = DeviceIdUtil::ExecutorForPlatformDeviceId(GPUMachineManager(), in GPUcudaMallocAllocator()
Dgpu_debug_allocator_test.cc42 return DeviceIdUtil::ExecutorForPlatformDeviceId(GPUMachineManager(), in ExecutorForPlatformGpuId()
Dgpu_device_test.cc35 DeviceIdUtil::ExecutorForPlatformDeviceId(GPUMachineManager(), gpu_id) in GetTotalGPUMemory()
46 DeviceIdUtil::ExecutorForPlatformDeviceId(GPUMachineManager(), gpu_id) in GetComputeCapability()
Dgpu_device.cc960 se::StreamExecutor* se = DeviceIdUtil::ExecutorForPlatformDeviceId( in SingleVirtualDeviceMemoryLimit()
1474 DeviceIdUtil::ExecutorForPlatformDeviceId(platform, platform_gpu_i) in GetPeerAccessMap()
1477 DeviceIdUtil::ExecutorForPlatformDeviceId(platform, platform_gpu_j) in GetPeerAccessMap()
1717 DeviceIdUtil::ExecutorForPlatformDeviceId(gpu_manager, platform_gpu_i) in EnablePeerAccess()
1720 DeviceIdUtil::ExecutorForPlatformDeviceId(gpu_manager, platform_gpu_j) in EnablePeerAccess()
Dgpu_bfc_allocator_test.cc67 DeviceIdUtil::ExecutorForPlatformDeviceId(GPUMachineManager(), gpu_id) in CreateVirtualMemorySubAllocator()
81 DeviceIdUtil::ExecutorForPlatformDeviceId(GPUMachineManager(), gpu_id) in CreateGPUMemAllocator()
Dgpu_process_state.cc87 auto executor = DeviceIdUtil::ExecutorForPlatformDeviceId(GPUMachineManager(), in CreateSubAllocator()