Home
last modified time | relevance | path

Searched refs:set_device_assignment (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/
Dexecutable_build_options.h106 ExecutableBuildOptions& set_device_assignment(
Dexecutable_build_options.cc92 ExecutableBuildOptions& ExecutableBuildOptions::set_device_assignment( in set_device_assignment() function in xla::ExecutableBuildOptions
/external/tensorflow/tensorflow/compiler/xla/
Dexecutable_run_options.cc95 ExecutableRunOptions& ExecutableRunOptions::set_device_assignment( in set_device_assignment() function in xla::ExecutableRunOptions
Dexecutable_run_options.h126 ExecutableRunOptions& set_device_assignment(
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dgpu_multistream_test.cc55 compile_options.executable_build_options.set_device_assignment( in TEST()
Dutils.cc134 build_options->set_device_assignment(device_assignment); in ParseDeviceAssignmentCompileOptions()
Dpjrt_stream_executor_client.cc1754 run_options.set_device_assignment(device_assignment.get()); in EnqueueExecution()
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_compiler.cc486 options.executable_build_options.set_device_assignment( in BuildXlaCompilerSubmodule()
553 &ExecutableBuildOptions::set_device_assignment) in BuildXlaCompilerSubmodule()
Doutfeed_receiver.cc404 compile_options.executable_build_options.set_device_assignment( in SendShutdownOutfeedHeader()
Doutfeed_receiver_test.cc43 compile_options.executable_build_options.set_device_assignment( in CompileAndExecute()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_compile_ops.cc175 build_options.set_device_assignment(device_assignment); in Compile()
Dxrt_execute_op.cc322 run_options.set_device_assignment( in RunExecutable()
/external/tensorflow/tensorflow/core/tpu/
Dtpu_execute.cc442 run_options.set_device_assignment(device_assignment); in TPUExecute()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_runner.cc452 run_options.set_device_assignment(device_assignment); in GetServiceRunOptionsForDevice()
Dservice.cc403 options.set_device_assignment(&device_assignment); in ExecuteParallelAndRegisterResult()
500 options.set_device_assignment(&device_assignment); in ExecuteAndRegisterResult()