Home
last modified time | relevance | path

Searched refs:has_device_assignment (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dutils.cc120 if (build_options->has_device_assignment()) { in ParseDeviceAssignmentCompileOptions()
128 if (!build_options->has_device_assignment()) { in ParseDeviceAssignmentCompileOptions()
/external/tensorflow/tensorflow/compiler/xla/client/
Dexecutable_build_options.h105 bool has_device_assignment() const { return device_assignment_.has_value(); } in has_device_assignment() function
Dexecutable_build_options.cc130 if (build_options.has_device_assignment()) { in CreateExecutionOptions()
Dlocal_client.cc366 if (options.has_device_assignment()) { in Compile()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module_util.cc114 execution_options->has_device_assignment()) { in CreateModuleConfig()
Dhlo_module.cc452 if (execution_options->has_device_assignment()) { in CreateModuleConfigFromShape()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_op_util.cc116 if (metadata.has_device_assignment()) { in CreateCompilationCacheKey()
Dtpu_compile_op_support.cc418 if (metadata->has_device_assignment()) { in CompileOpMetadataFromContext()
Dtpu_execute_op.cc732 if (executable.has_device_assignment()) { in DoWork()
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_compiler.cc478 return options.executable_build_options.has_device_assignment() in BuildXlaCompilerSubmodule()
548 return options.has_device_assignment() in BuildXlaCompilerSubmodule()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_compile_ops.cc169 if (config.has_device_assignment()) { in Compile()
Dtpu_execute_op.cc199 if (executable.has_device_assignment()) { in RunExecutable()
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.cc815 if (options.has_device_assignment()) { in Compile()