HomeSort by relevance Sort by last modified time
    Searched refs:execution_options_ (Results 1 - 16 of 16) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/tests/
check_execution_arity_test.cc 57 computation, {param0_data.get(), param1_data.get()}, &execution_options_);
61 client_->Execute(computation, {param0_data.get()}, &execution_options_);
69 client_->Execute(computation, {}, &execution_options_);
98 computation, {f32_data.get(), f32_4_data.get()}, &execution_options_);
103 &execution_options_);
112 &execution_options_);
121 &execution_options_);
execution_profile_test.cc 44 &execution_options_, &execution_profile));
54 execution_options_.debug_options().xla_hlo_profile();
reduce_precision_test.cc 272 auto reduce_precision_pass = execution_options_.mutable_debug_options()
298 auto reduce_precision_pass = execution_options_.mutable_debug_options()
322 auto reduce_precision_pass = execution_options_.mutable_debug_options()
347 auto reduce_precision_pass = execution_options_.mutable_debug_options()
372 auto reduce_precision_pass = execution_options_.mutable_debug_options()
client_test.cc 49 ExecutionOptions execution_options = execution_options_;
79 ExecutionOptions execution_options = execution_options_;
132 ExecutionOptions options = execution_options_;
replay_test.cc 63 ->ExecuteAndTransfer(replayed, /*arguments=*/{}, &execution_options_)
103 &execution_options_)
145 ->ExecuteAndTransfer(replayed, /*arguments=*/{}, &execution_options_)
client_library_test_base.cc 52 execution_options_(CreateDefaultExecutionOptions()) {
60 execution_options_.mutable_debug_options()->add_xla_disable_hlo_passes(
65 : execution_options_(CreateDefaultExecutionOptions()) {
69 execution_options_.mutable_debug_options()->add_xla_disable_hlo_passes(
82 return client_->Execute(computation, arguments, &execution_options_);
89 ExecutionOptions execution_options = execution_options_;
129 client_->ExecuteAndTransfer(computation, arguments, &execution_options_);
296 if (execution_options_.debug_options().xla_test_all_output_layouts()) {
300 if (execution_options_.debug_options().xla_test_all_input_layouts()) {
348 if (execution_options_.debug_options().xla_test_all_output_layouts())
    [all...]
prng_test.cc 206 ExecutionOptions execution_options = execution_options_;
236 ExecutionOptions execution_options1 = execution_options_;
239 ExecutionOptions execution_options2 = execution_options_;
271 &execution_options_));
274 &execution_options_));
compilation_cache_test.cc 49 /*execution_options=*/&execution_options_,
65 &execution_options_, &execution_profile)
deallocation_test.cc 43 client_->Execute(computation, arguments, &execution_options_)
client_library_test_base.h 77 execution_options_.mutable_debug_options()->set_xla_enable_fast_math(
81 void SetSeed(uint64 seed) { execution_options_.set_seed(seed); }
86 return execution_options_.mutable_debug_options();
92 // execution options. Modify execution_options_ in your test if you want to
384 ExecutionOptions execution_options_; member in class:xla::ClientLibraryTestBase
deconstruct_tuple_test.cc 49 client_->Execute(computation, arguments, &execution_options_)
batch_normalization_test.cc 533 execution_options_.mutable_debug_options()->clear_xla_disable_hlo_passes();
639 execution_options_.mutable_debug_options()->clear_xla_disable_hlo_passes();
    [all...]
reshape_test.cc 649 ExecutionOptions execution_options = execution_options_;
    [all...]
scalar_computations_test.cc 392 &execution_options_)
433 &execution_options_)
    [all...]
convolution_test.cc 552 execution_options_.mutable_debug_options()->add_xla_disable_hlo_passes(
while_test.cc     [all...]

Completed in 132 milliseconds