Home
last modified time | relevance | path

Searched refs:static_device_assignment (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dcompile_only_service.cc80 options.static_device_assignment().replica_count()); in CompileAheadOfTime()
87 options.static_device_assignment().computation_count()); in CompileAheadOfTime()
91 TF_RETURN_IF_ERROR(options.static_device_assignment().Serialize( in CompileAheadOfTime()
Dhlo_module_config.h181 const DeviceAssignment& static_device_assignment() const { in static_device_assignment() function
Dcompiler.h99 const DeviceAssignment& static_device_assignment() const { in static_device_assignment() function
Dhlo_module.cc458 CHECK_EQ(module_config.static_device_assignment().replica_count(), in CreateModuleConfigFromShape()
462 CHECK_EQ(module_config.static_device_assignment().computation_count(), in CreateModuleConfigFromShape()
/external/tensorflow/tensorflow/stream_executor/tpu/
Dc_api_conversions.cc428 config.static_device_assignment().Serialize(&dev_proto).IgnoreError(); in ToC()
429 hlo_config.static_device_assignment = in ToC()
465 c_config.static_device_assignment)); in FromC()
485 c_config->static_device_assignment); in Free()
Dc_api_decl.h284 TpuSerializedProto static_device_assignment; member
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_execute_op.cc323 &executable->executable()->module_config().static_device_assignment()); in RunExecutable()