Searched refs:set_device_assignment (Results 1 – 15 of 15) sorted by relevance
106 ExecutableBuildOptions& set_device_assignment(
92 ExecutableBuildOptions& ExecutableBuildOptions::set_device_assignment( in set_device_assignment() function in xla::ExecutableBuildOptions
95 ExecutableRunOptions& ExecutableRunOptions::set_device_assignment( in set_device_assignment() function in xla::ExecutableRunOptions
126 ExecutableRunOptions& set_device_assignment(
55 compile_options.executable_build_options.set_device_assignment( in TEST()
134 build_options->set_device_assignment(device_assignment); in ParseDeviceAssignmentCompileOptions()
1754 run_options.set_device_assignment(device_assignment.get()); in EnqueueExecution()
486 options.executable_build_options.set_device_assignment( in BuildXlaCompilerSubmodule()553 &ExecutableBuildOptions::set_device_assignment) in BuildXlaCompilerSubmodule()
404 compile_options.executable_build_options.set_device_assignment( in SendShutdownOutfeedHeader()
43 compile_options.executable_build_options.set_device_assignment( in CompileAndExecute()
175 build_options.set_device_assignment(device_assignment); in Compile()
322 run_options.set_device_assignment( in RunExecutable()
442 run_options.set_device_assignment(device_assignment); in TPUExecute()
452 run_options.set_device_assignment(device_assignment); in GetServiceRunOptionsForDevice()
403 options.set_device_assignment(&device_assignment); in ExecuteParallelAndRegisterResult()500 options.set_device_assignment(&device_assignment); in ExecuteAndRegisterResult()