Searched refs:set_device_assignment (Results 1 – 19 of 19) sorted by relevance
153 output.set_device_assignment(*assignment); in ExecutableBuildOptionsFromProto()162 ExecutableBuildOptions& ExecutableBuildOptions::set_device_assignment( in set_device_assignment() function in xla::ExecutableBuildOptions
121 ExecutableBuildOptions& set_device_assignment(
95 ExecutableRunOptions& ExecutableRunOptions::set_device_assignment( in set_device_assignment() function in xla::ExecutableRunOptions
126 ExecutableRunOptions& set_device_assignment(
56 compile_options.executable_build_options.set_device_assignment( in TEST()
136 build_options->set_device_assignment(device_assignment); in ParseDeviceAssignmentCompileOptions()
90 options.executable_build_options.set_device_assignment(assignment); in MakeIncrementProgram()
1459 run_options.set_device_assignment(device_assignment.get()); in ExecuteHelper()
1981 run_options.set_device_assignment(device_assignment.get()); in EnqueueExecution()
176 build_options.set_device_assignment(device_assignment); in Compile()
324 run_options.set_device_assignment( in RunExecutable()
251 run_options.set_device_assignment(&device_assignment); in ResolveDeviceAssignment()
641 options.executable_build_options.set_device_assignment( in BuildXlaCompilerSubmodule()716 &ExecutableBuildOptions::set_device_assignment) in BuildXlaCompilerSubmodule()
43 compile_options.executable_build_options.set_device_assignment( in CompileAndExecute()
411 compile_options.executable_build_options.set_device_assignment( in SendShutdownOutfeedHeader()
446 run_options.set_device_assignment(device_assignment); in TPUExecute()
433 options.set_device_assignment(&device_assignment); in ExecuteParallelAndRegisterResult()531 options.set_device_assignment(&device_assignment); in ExecuteAndRegisterResult()
482 run_options.set_device_assignment(device_assignment); in GetServiceRunOptionsForDevice()
224 ret.executable_build_options.set_device_assignment(*device_assignment); in ConvertCCompileOptionstoCppCompileOptions()