| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | service.cc | 391 DeviceAssignment device_assignment(options_.number_of_replicas(), in ExecuteParallelAndRegisterResult() local
515 DeviceAssignment device_assignment(options_.number_of_replicas(), in ExecuteAndRegisterResult() local
|
| H A D | compiler.h | 124 void set_static_device_assignment(const DeviceAssignment& device_assignment) { in set_static_device_assignment()
|
| H A D | hlo_module_config.h | 233 void set_static_device_assignment(const DeviceAssignment& device_assignment) { in set_static_device_assignment()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | all_reduce_blueconnect_test.cc | 38 DeviceAssignment device_assignment(replica_count, /*computation_count=*/1); in SetModuleConfig() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ |
| H A D | device_assignment.py | 332 def device_assignment( function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| H A D | utils.cc | 120 std::shared_ptr<DeviceAssignment>* device_assignment) { in ParseDeviceAssignmentCompileOptions()
|
| H A D | pjrt_c_api_client.h | 443 const DeviceAssignment& device_assignment() const override { in device_assignment() function
|
| H A D | tfrt_cpu_pjrt_client.h | 496 const DeviceAssignment& device_assignment() const override { in device_assignment() function
|
| /aosp_15_r20/packages/modules/Virtualization/guest/pvmfw/src/ |
| D | main.rs | 25 mod device_assignment; module
|
| D | fdt.rs | 1003 device_assignment: Option<DeviceAssignmentInfo>, field
1151 let device_assignment = match vm_dtbo { in parse_device_tree() localVariable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | xla_helpers.cc | 150 xla::DeviceAssignment& device_assignment, in ResolveDeviceAssignment()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/ |
| H A D | grpc_tpu_driver.cc | 412 const xla::DeviceAssignmentProto& device_assignment, in ExecuteProgram()
901 const xla::DeviceAssignmentProto& device_assignment, in ExecuteProgram()
|
| H A D | tpu_service.proto | 94 optional xla.DeviceAssignmentProto device_assignment = 4; field
|
| H A D | direct_tpu_driver.cc | 432 const xla::DeviceAssignmentProto& device_assignment, in ExecuteProgram()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/ |
| H A D | xrt.proto | 50 DeviceAssignment device_assignment = 6; field
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/ |
| H A D | outfeed_receiver.cc | 409 DeviceAssignment device_assignment(1, 1); in SendShutdownOutfeedHeader() local
|
| H A D | outfeed_receiver_test.cc | 41 DeviceAssignment device_assignment(1, 1); in CompileAndExecute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| H A D | tpu_client.h | 371 const DeviceAssignment& device_assignment() const { in device_assignment() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/ |
| H A D | tpu_execute.cc | 422 TpuNodeContext* node_context, xla::DeviceAssignment* device_assignment, in TPUExecute()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| H A D | tpu_execute_op.cc | 199 std::unique_ptr<xla::DeviceAssignment> device_assignment; in RunExecutable() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
| H A D | c_api_conversions.cc | 518 auto device_assignment = xla::DeviceAssignment::Deserialize( in FromC() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/kernels/ |
| H A D | xla_ops.cc | 355 xla::DeviceAssignment device_assignment; in ComputeAsync() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| H A D | tpu_rewrite_device_util.cc | 379 xla::DeviceAssignment device_assignment(num_replicas, num_cores_per_replica); in GetGeneralTPUExecutionDeviceAssignment() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| H A D | distributed_tpu_rewrite_pass.cc | 1655 xla::Array2D<tpu::TpuCoreLocationExternal>* device_assignment) { in ParseDeviceAssignmentAttr()
1738 const xla::Array2D<tpu::TpuCoreLocationExternal>& device_assignment, in BuildGeneralDeviceAssignment()
1864 xla::Array2D<tpu::TpuCoreLocationExternal> device_assignment; in BuildDeviceAssignment() local
4399 std::vector<int> device_assignment; in GetDeviceTopology() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | tpu_execute_op.cc | 726 std::unique_ptr<xla::DeviceAssignment> device_assignment; in DoWork() local
|