HomeSort by relevance Sort by last modified time
    Searched refs:ExecutableRunOptions (Results 1 - 25 of 35) sorted by null

1 2

  /external/tensorflow/tensorflow/compiler/xla/
executable_run_options.cc 20 ExecutableRunOptions& ExecutableRunOptions::set_device_ordinal(
26 int ExecutableRunOptions::device_ordinal() const { return device_ordinal_; }
28 ExecutableRunOptions& ExecutableRunOptions::set_allocator(
34 DeviceMemoryAllocator* ExecutableRunOptions::allocator() const {
38 ExecutableRunOptions& ExecutableRunOptions::set_stream(
44 stream_executor::Stream* ExecutableRunOptions::stream() const {
48 ExecutableRunOptions& ExecutableRunOptions::set_host_to_device_stream
    [all...]
executable_run_options.h 19 // These classes are forward declared so that ExecutableRunOptions can be linked
39 class ExecutableRunOptions {
42 ExecutableRunOptions& set_allocator(DeviceMemoryAllocator* allocator);
50 ExecutableRunOptions& set_device_ordinal(int device_ordinal);
56 ExecutableRunOptions& set_stream(stream_executor::Stream* stream);
62 ExecutableRunOptions& set_host_to_device_stream(
68 ExecutableRunOptions& set_intra_op_thread_pool(
74 ExecutableRunOptions& set_execution_profile(ExecutionProfile* profile);
76 ExecutableRunOptions& set_device_assignment(
80 ExecutableRunOptions& set_rng_seed(int rng_seed)
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/
service_executable_run_options.h 33 : ServiceExecutableRunOptions(ExecutableRunOptions()) {}
35 explicit ServiceExecutableRunOptions(ExecutableRunOptions run_options,
40 // Returns reference or pointer to `ExecutableRunOptions` member.
41 const ExecutableRunOptions& run_options() const { return run_options_; }
42 ExecutableRunOptions* mutable_run_options() { return &run_options_; }
44 // Delegate to `ExecutableRunOptions` member.
58 ExecutableRunOptions run_options_;
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_runtime.h 107 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length,
123 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length,
129 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length,
142 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length,
runtime_fft.cc 32 const xla::ExecutableRunOptions* run_options =
33 static_cast<const xla::ExecutableRunOptions*>(run_options_ptr);
runtime_conv2d.cc 35 const xla::ExecutableRunOptions* run_options =
36 static_cast<const xla::ExecutableRunOptions*>(run_options_ptr);
54 const xla::ExecutableRunOptions* run_options =
55 static_cast<const xla::ExecutableRunOptions*>(run_options_ptr);
runtime_fork_join.cc 74 const xla::ExecutableRunOptions* run_options =
75 static_cast<const xla::ExecutableRunOptions*>(run_options_ptr);
cpu_runtime.cc 109 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length,
133 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length,
152 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length,
176 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length,
runtime_matmul_mkl.cc 81 const xla::ExecutableRunOptions* run_options =
82 static_cast<const xla::ExecutableRunOptions*>(run_options_ptr);
96 const xla::ExecutableRunOptions* run_options =
97 static_cast<const xla::ExecutableRunOptions*>(run_options_ptr);
cpu_executable.h 78 void (*)(void* /*result*/, const ExecutableRunOptions* /*run_options*/,
120 Status ExecuteComputeFunction(const ExecutableRunOptions* run_options,
runtime_matmul.cc 42 const xla::ExecutableRunOptions* run_options =
43 static_cast<const xla::ExecutableRunOptions*>(run_options_ptr);
cpu_runtime_test.cc 106 ExecutableRunOptions run_options;
cpu_executable.cc 137 const ExecutableRunOptions* run_options,
146 // run_options: the ExecutableRunOptions object.
  /external/tensorflow/tensorflow/compiler/xla/tests/
local_client_aot_test.cc 24 extern "C" void SumAndDouble(float* out, xla::ExecutableRunOptions* options,
43 xla::ExecutableRunOptions run_options;
local_client_test_base.h 100 const ExecutableRunOptions& run_options);
109 const ExecutableRunOptions& run_options);
116 ExecutableRunOptions DefaultExecutableRunOptions() const;
local_client_allocation_test.cc 54 ExecutableRunOptions options;
93 ExecutableRunOptions().set_device_ordinal(d).set_allocator(allocator));
local_client_test_base.cc 150 ExecutableRunOptions LocalClientTestBase::DefaultExecutableRunOptions() const {
151 ExecutableRunOptions run_options;
169 const ExecutableRunOptions& run_options) {
185 const ExecutableRunOptions& run_options) {
multiple_devices_on_host_test.cc 40 xla::ExecutableRunOptions execute_options;
  /external/tensorflow/tensorflow/compiler/xla/client/
local_client.h 44 ExecutableRunOptions run_options);
63 // The given ExecutableRunOptions override any values from TF_XLA_FLAGS
67 const ExecutableRunOptions& run_options, const Backend& backend);
local_client.cc 56 const ExecutableRunOptions& run_options, const Backend& backend) {
100 "ExecutableRunOptions; the stream determines the device ordinal");
145 ExecutableRunOptions run_options) {
165 // ExecutableRunOptions.eigen_intra_op_thread_pool.
  /external/tensorflow/tensorflow/compiler/tf2xla/
xla_compiled_cpu_function.h 53 const xla::ExecutableRunOptions* run_options,
343 xla::ExecutableRunOptions run_options_;
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
buffer_comparator.cc 141 ExecutableRunOptions run_options;
  /external/tensorflow/tensorflow/compiler/jit/
xla_compile_on_demand_op.cc 75 xla::ExecutableRunOptions run_options;
  /external/tensorflow/tensorflow/compiler/jit/kernels/
xla_ops.cc 360 xla::ExecutableRunOptions run_options;
534 xla::ExecutableRunOptions run_options;
  /external/tensorflow/tensorflow/compiler/xrt/kernels/
xrt_execute_op.cc 207 xla::ExecutableRunOptions run_options;

Completed in 549 milliseconds

1 2