Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dnccl_collective_thunk.cc200 GlobalDeviceId global_device_id = in GetDeviceString() local
H A Djitrt_custom_calls.cc2026 StatusOr<GlobalDeviceId> global_device_id = params.GetGlobalDeviceId(); in operator ()() local
2102 StatusOr<GlobalDeviceId> global_device_id = params.GetGlobalDeviceId(); in operator ()() local
2143 StatusOr<GlobalDeviceId> global_device_id = params.GetGlobalDeviceId(); in operator ()() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/distributed/
H A Dprotocol.proto44 int32 global_device_id = 4; // Globally unique ID number. field
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_configuration_ops.cc206 int32_t global_device_id = host_ordinal_to_global_device_id(j); in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dgpu_device.cc364 GlobalDeviceId global_device_id(device_proto.global_device_id()); in BuildDistributedDevices() local
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/compiler/xla/pjrt/distributed/
Dprotocol.pb.h2742 inline ::int32_t DeviceProto::global_device_id() const { in global_device_id() function