Home
last modified time | relevance | path

Searched defs:TfDeviceSpec (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_device.cc1454 struct TfDeviceSpec { in CreateDevices() struct
1455 PlatformDeviceId platform_device_id; in CreateDevices()
1456 int64_t memory_limit_bytes; in CreateDevices()
1457 int index; // Index in the concatenated virtual device configuration list. in CreateDevices()
1458 int device_ordinal; // Ordinal for determining the tf device id. in CreateDevices()
1459 TfDeviceSpec(PlatformDeviceId platform_device_id, in CreateDevices() argument