Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/pjrt/distributed/
Dprotocol.proto41 int32 global_device_id = 4; // Globally unique ID number. field
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_configuration_ops.cc184 int32_t global_device_id = host_ordinal_to_global_device_id(j); in Compute() local
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dgpu_device.cc272 GlobalDeviceId global_device_id(device_proto.global_device_id()); in BuildDistributedDevices() local