| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | check_execution_arity_test.cc | 57 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()
|
| D | client_library_test_base.cc | 64 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 …]
|
| D | prng_test.cc | 216 ExecutionOptions execution_options = execution_options_; in XLA_TEST_F()
245 ExecutionOptions execution_options1 = execution_options_; in XLA_TEST_F()
248 ExecutionOptions execution_options2 = execution_options_; in XLA_TEST_F()
280 &execution_options_)); in XLA_TEST_F()
283 &execution_options_)); in XLA_TEST_F()
309 ExecutionOptions execution_options = execution_options_; in XLA_TEST_F()
|
| D | execution_profile_test.cc | 44 &execution_options_, &execution_profile)); in XLA_TEST_F()
54 execution_options_.debug_options().xla_hlo_profile(); in XLA_TEST_F()
|
| D | replay_test.cc | 64 ->ExecuteAndTransfer(replayed, /*arguments=*/{}, &execution_options_) in TEST_F()
105 &execution_options_) in XLA_TEST_F()
144 ->ExecuteAndTransfer(replayed, /*arguments=*/{}, &execution_options_) in TEST_F()
|
| D | client_test.cc | 50 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()
|
| D | compilation_cache_test.cc | 48 /*execution_options=*/&execution_options_, in ExecuteComputationR0F32()
64 &execution_options_, &execution_profile) in ExecuteComputationR2F32()
|
| D | deallocation_test.cc | 42 client_->Execute(computation, arguments, &execution_options_) in ExecuteAndCheckTransfer()
|
| D | client_library_test_base.h | 78 auto* opts = execution_options_.mutable_debug_options(); in SetFastMathDisabled()
84 void SetSeed(uint64 seed) { execution_options_.set_seed(seed); } in SetSeed()
89 return execution_options_.mutable_debug_options(); in mutable_debug_options()
422 ExecutionOptions execution_options_; variable
|
| D | deconstruct_tuple_test.cc | 48 client_->Execute(computation, arguments, &execution_options_) in ExecuteAndCheckTransfer()
|
| D | batch_normalization_test.cc | 641 execution_options_.mutable_debug_options()->clear_xla_disable_hlo_passes(); in XLA_TEST_P()
747 execution_options_.mutable_debug_options()->clear_xla_disable_hlo_passes(); in XLA_TEST_P()
935 execution_options_.mutable_debug_options()->clear_xla_disable_hlo_passes(); in XLA_TEST_P()
|
| D | scalar_computations_test.cc | 397 &execution_options_) in XLA_TEST_F()
439 &execution_options_) in XLA_TEST_F()
|
| D | reshape_test.cc | 651 ExecutionOptions execution_options = execution_options_; in XLA_TEST_P()
807 ExecutionOptions execution_options = execution_options_; in XLA_TEST_P()
|
| D | dot_operation_test.cc | 437 execution_options_.mutable_debug_options()->add_xla_disable_hlo_passes( in ParametricDotTestWithoutLayoutAssignment()
439 execution_options_.mutable_debug_options()->add_xla_disable_hlo_passes( in ParametricDotTestWithoutLayoutAssignment()
443 execution_options_.mutable_debug_options()->add_xla_disable_hlo_passes( in ParametricDotTestWithoutLayoutAssignment()
|
| D | dynamic_ops_test.cc | 311 execution_options_.mutable_debug_options()->add_xla_disable_hlo_passes( in TestR0()
|
| D | convolution_test.cc | 1511 execution_options_.mutable_debug_options()->add_xla_disable_hlo_passes( in XLA_TEST_P()
1577 execution_options_.mutable_debug_options()->add_xla_disable_hlo_passes( in XLA_TEST_F()
|
| D | while_test.cc | 942 ExecutionOptions execution_options = execution_options_; in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_infeed_test.cc | 154 result = client_->Execute(computation, {}, &execution_options_) in TEST_F()
270 result = client_->Execute(computation, {}, &execution_options_) in TEST_F()
|