Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_executable_run_options.cc70 int64_t local_device_ordinal = stream->parent()->device_ordinal(); in GetGlobalDeviceId() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/distributed/
H A Dprotocol.proto36 int32 local_device_ordinal = 1; field
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/
H A Dtpu_ops_c_api.h633 int32_t local_device_ordinal; member
/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.h2632 inline ::int32_t DeviceProto::local_device_ordinal() const { in local_device_ordinal() function