Home
last modified time | relevance | path

Searched refs:executable_build_options (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dgpu_multistream_test.cc49 compile_options.executable_build_options.mutable_debug_options() in TEST()
51 compile_options.executable_build_options.mutable_debug_options() in TEST()
55 compile_options.executable_build_options.set_device_assignment( in TEST()
DBUILD143 "//tensorflow/compiler/xla/client:executable_build_options",
164 "//tensorflow/compiler/xla/client:executable_build_options",
193 "//tensorflow/compiler/xla/client:executable_build_options",
341 "//tensorflow/compiler/xla/client:executable_build_options",
Dpjrt_client.h124 ExecutableBuildOptions executable_build_options; member
Dpjrt_stream_executor_client.cc2119 ExecutableBuildOptions& build_options = options.executable_build_options; in Compile()
2131 options.compile_portable_executable, &options.executable_build_options, in Compile()
2145 options.argument_layouts, &options.executable_build_options, in Compile()
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_compiler.cc443 options.executable_build_options.mutable_debug_options(); in BuildXlaCompilerSubmodule()
453 &CompileOptions::executable_build_options) in BuildXlaCompilerSubmodule()
461 return options.executable_build_options.num_replicas(); in BuildXlaCompilerSubmodule()
464 options.executable_build_options.set_num_replicas(num_replicas); in BuildXlaCompilerSubmodule()
469 return options.executable_build_options.num_partitions(); in BuildXlaCompilerSubmodule()
472 options.executable_build_options.set_num_partitions(num_partitions); in BuildXlaCompilerSubmodule()
478 return options.executable_build_options.has_device_assignment() in BuildXlaCompilerSubmodule()
480 options.executable_build_options in BuildXlaCompilerSubmodule()
486 options.executable_build_options.set_device_assignment( in BuildXlaCompilerSubmodule()
Doutfeed_receiver.cc400 compile_options.executable_build_options.set_num_replicas(1); in SendShutdownOutfeedHeader()
401 compile_options.executable_build_options.set_num_partitions(1); in SendShutdownOutfeedHeader()
404 compile_options.executable_build_options.set_device_assignment( in SendShutdownOutfeedHeader()
Doutfeed_receiver_test.cc39 compile_options.executable_build_options.set_num_replicas(1); in CompileAndExecute()
40 compile_options.executable_build_options.set_num_partitions(1); in CompileAndExecute()
43 compile_options.executable_build_options.set_device_assignment( in CompileAndExecute()
DBUILD361 "//tensorflow/compiler/xla/client:executable_build_options",
474 "//tensorflow/compiler/xla/client:executable_build_options",
/external/tensorflow/tensorflow/compiler/xla/client/
DBUILD93 name = "executable_build_options",
94 srcs = ["executable_build_options.cc"],
95 hdrs = ["executable_build_options.h"],
117 ":executable_build_options",
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
DBUILD29 "//tensorflow/compiler/xla/client:executable_build_options",
Dtpu_client_extension.cc137 &options.executable_build_options, client, in PYBIND11_MODULE()
/external/tensorflow/tensorflow/compiler/xla/service/
DBUILD1006 "//tensorflow/compiler/xla/client:executable_build_options",