Home
last modified time | relevance | path

Searched defs:device_assignment (Results 51 – 60 of 60) sorted by relevance

123

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_module.cc352 DeviceAssignmentProto device_assignment; in ToProto() local
H A Dhlo.proto521 DeviceAssignmentProto device_assignment = 15; field
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
H A Drecording_tpu_driver.cc415 const xla::DeviceAssignmentProto& device_assignment, in ExecuteProgram()
H A Dpod_tpu_driver.cc656 const xla::DeviceAssignmentProto& device_assignment, in ExecuteProgram()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dxla.proto476 DeviceAssignmentProto device_assignment = 7; field
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dxla_compiler.cc640 const DeviceAssignment& device_assignment) { in BuildXlaCompilerSubmodule()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_runtime.cc697 const xla::DeviceAssignment& device_assignment = in GetRendezvousKey() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A Dc_api_decl.h164 TpuSerializedProto device_assignment; member
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/
H A Dtpu_ops_c_api.h234 XLA_DeviceAssignment* device_assignment; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/c/
H A Dpjrt_c_api.h329 const char* device_assignment; // Serialized device assignment. member

123