/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...] |