Home
last modified time | relevance | path

Searched refs:execution_options_ (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dcheck_execution_arity_test.cc57 computation, {param0_data.get(), param1_data.get()}, &execution_options_); in TEST_F()
61 client_->Execute(computation, {param0_data.get()}, &execution_options_); in TEST_F()
69 client_->Execute(computation, {}, &execution_options_); in TEST_F()
98 computation, {f32_data.get(), f32_4_data.get()}, &execution_options_); in XLA_TEST_F()
103 &execution_options_); in XLA_TEST_F()
112 &execution_options_); in XLA_TEST_F()
121 &execution_options_); in XLA_TEST_F()
Dclient_library_test_base.cc64 execution_options_(CreateDefaultExecutionOptions()) { in ClientLibraryTestBase()
77 execution_options_.mutable_debug_options()->add_xla_disable_hlo_passes( in ClientLibraryTestBase()
80 execution_options_.mutable_debug_options() in ClientLibraryTestBase()
85 : execution_options_(CreateDefaultExecutionOptions()) { in ClientLibraryTestBase()
94 execution_options_.mutable_debug_options()->add_xla_disable_hlo_passes( in ClientLibraryTestBase()
97 execution_options_.mutable_debug_options() in ClientLibraryTestBase()
109 return client_->Execute(computation, arguments, &execution_options_); in Execute()
115 ExecutionOptions execution_options = execution_options_; in ExecuteAndTransfer()
135 ExecutionOptions execution_options = execution_options_; in ExecuteAndTransferReference()
154 client_->ExecuteAndTransfer(computation, arguments, &execution_options_); in ExecuteToString()
[all …]
Dreduce_precision_test.cc270 auto reduce_precision_pass = execution_options_.mutable_debug_options() in XLA_TEST_F()
296 auto reduce_precision_pass = execution_options_.mutable_debug_options() in XLA_TEST_F()
320 auto reduce_precision_pass = execution_options_.mutable_debug_options() in XLA_TEST_F()
345 auto reduce_precision_pass = execution_options_.mutable_debug_options() in XLA_TEST_F()
370 auto reduce_precision_pass = execution_options_.mutable_debug_options() in XLA_TEST_F()
Dprng_test.cc207 ExecutionOptions execution_options = execution_options_; in XLA_TEST_F()
235 ExecutionOptions execution_options1 = execution_options_; in XLA_TEST_F()
238 ExecutionOptions execution_options2 = execution_options_; in XLA_TEST_F()
270 &execution_options_)); in XLA_TEST_F()
273 &execution_options_)); in XLA_TEST_F()
299 ExecutionOptions execution_options = execution_options_; in XLA_TEST_F()
Dreplay_test.cc64 ->ExecuteAndTransfer(replayed, /*arguments=*/{}, &execution_options_) in TEST_F()
105 &execution_options_) in XLA_TEST_F()
144 ->ExecuteAndTransfer(replayed, /*arguments=*/{}, &execution_options_) in TEST_F()
Dclient_test.cc50 ExecutionOptions execution_options = execution_options_; in XLA_TEST_F()
80 ExecutionOptions execution_options = execution_options_; in XLA_TEST_F()
136 ExecutionOptions options = execution_options_; in XLA_TEST_F()
Dexecution_profile_test.cc44 &execution_options_, &execution_profile)); in XLA_TEST_F()
54 execution_options_.debug_options().xla_hlo_profile(); in XLA_TEST_F()
Dcompilation_cache_test.cc48 /*execution_options=*/&execution_options_, in ExecuteComputationR0F32()
64 &execution_options_, &execution_profile) in ExecuteComputationR2F32()
Dclient_library_test_base.h77 auto* opts = execution_options_.mutable_debug_options(); in SetFastMathDisabled()
82 void SetSeed(uint64 seed) { execution_options_.set_seed(seed); } in SetSeed()
87 return execution_options_.mutable_debug_options(); in mutable_debug_options()
390 ExecutionOptions execution_options_; variable
Ddeallocation_test.cc42 client_->Execute(computation, arguments, &execution_options_) in ExecuteAndCheckTransfer()
Dbatch_normalization_test.cc543 execution_options_.mutable_debug_options()->clear_xla_disable_hlo_passes(); in XLA_TEST_P()
649 execution_options_.mutable_debug_options()->clear_xla_disable_hlo_passes(); in XLA_TEST_P()
837 execution_options_.mutable_debug_options()->clear_xla_disable_hlo_passes(); in XLA_TEST_P()
Ddeconstruct_tuple_test.cc48 client_->Execute(computation, arguments, &execution_options_) in ExecuteAndCheckTransfer()
Dscalar_computations_test.cc387 &execution_options_) in XLA_TEST_F()
429 &execution_options_) in XLA_TEST_F()
Dreshape_test.cc618 ExecutionOptions execution_options = execution_options_; in XLA_TEST_P()
768 ExecutionOptions execution_options = execution_options_; in XLA_TEST_P()
Ddot_operation_test.cc396 execution_options_.mutable_debug_options()->add_xla_disable_hlo_passes( in ParametricDotTestWithoutLayoutAssignment()
400 execution_options_.mutable_debug_options()->add_xla_disable_hlo_passes( in ParametricDotTestWithoutLayoutAssignment()
Ddynamic_ops_test.cc310 execution_options_.mutable_debug_options()->add_xla_disable_hlo_passes( in TestR0()
Dconvolution_test.cc1679 execution_options_.mutable_debug_options()->add_xla_disable_hlo_passes( in XLA_TEST_P()
1891 execution_options_.mutable_debug_options()->add_xla_disable_hlo_passes( in XLA_TEST_F()
Dwhile_test.cc942 ExecutionOptions execution_options = execution_options_; in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_infeed_test.cc154 result = client_->Execute(computation, {}, &execution_options_) in TEST_F()
270 result = client_->Execute(computation, {}, &execution_options_) in TEST_F()