Home
last modified time | relevance | path

Searched refs:ExecuteAndTransfer (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/tests/
Dbroadcast_test.cc47 auto result = ExecuteAndTransfer(std::move(hlo_module), {}); in XLA_TEST_F()
63 auto result = ExecuteAndTransfer(std::move(hlo_module), {}); in XLA_TEST_F()
86 auto result = ExecuteAndTransfer(std::move(hlo_module), {}); in XLA_TEST_F()
107 auto result = ExecuteAndTransfer(std::move(hlo_module), {}); in XLA_TEST_F()
126 auto result = ExecuteAndTransfer(std::move(hlo_module), {}); in XLA_TEST_F()
143 auto result = ExecuteAndTransfer(std::move(hlo_module), {}); in XLA_TEST_F()
163 auto result = ExecuteAndTransfer(std::move(hlo_module), {}); in TEST_F()
188 auto result = ExecuteAndTransfer(std::move(hlo_module), {}); in TEST_F()
219 auto result = ExecuteAndTransfer(std::move(hlo_module), {}); in XLA_TEST_F()
236 auto result = ExecuteAndTransfer(std::move(hlo_module), {}); in TEST_F()
[all …]
Dprng_test.cc60 ExecuteAndTransfer(&builder, /*arguments=*/{}).ConsumeValueOrDie(); in UniformTest()
151 ExecuteAndTransfer(&builder, /*arguments=*/{}).ConsumeValueOrDie(); in UniformChiSquared()
210 auto actual, client_->ExecuteAndTransfer( in XLA_TEST_F()
245 result1, client_->ExecuteAndTransfer(computation, /*arguments=*/{}, in XLA_TEST_F()
253 result2, client_->ExecuteAndTransfer(computation, /*arguments=*/{}, in XLA_TEST_F()
256 result3, client_->ExecuteAndTransfer(computation, /*arguments=*/{}, in XLA_TEST_F()
266 result4, client_->ExecuteAndTransfer(computation, /*arguments=*/{}, in XLA_TEST_F()
269 result5, client_->ExecuteAndTransfer(computation, /*arguments=*/{}, in XLA_TEST_F()
272 result6, client_->ExecuteAndTransfer(computation, /*arguments=*/{}, in XLA_TEST_F()
306 result_tuple, client_->ExecuteAndTransfer(computation, /*arguments=*/{}, in XLA_TEST_F()
[all …]
Dfusion_test.cc131 auto actual = ExecuteAndTransfer(std::move(hlo_module), {}); in TestElementwise2D()
242 ExecuteAndTransfer(std::move(hlo_module), {}), ErrorSpec(1e-4))); in XLA_TEST_F()
268 ExecuteAndTransfer(std::move(hlo_module), {}), ErrorSpec(1e-4))); in XLA_TEST_F()
298 auto result = ExecuteAndTransfer(std::move(hlo_module), {}); in XLA_TEST_F()
328 ExecuteAndTransfer(std::move(hlo_module), {}), ErrorSpec(1e-4))); in XLA_TEST_F()
343 ExecuteAndTransfer(std::move(hlo_module), {}))); in XLA_TEST_F()
358 ExecuteAndTransfer(std::move(hlo_module), {}))); in XLA_TEST_F()
373 ExecuteAndTransfer(std::move(hlo_module), {}))); in XLA_TEST_F()
388 ExecuteAndTransfer(std::move(hlo_module), {}))); in XLA_TEST_F()
403 ExecuteAndTransfer(std::move(hlo_module), {}))); in XLA_TEST_F()
[all …]
Dclient_library_test_base.cc112 StatusOr<Literal> ClientLibraryTestBase::ExecuteAndTransfer( in ExecuteAndTransfer() function in xla::ClientLibraryTestBase
120 return client_->ExecuteAndTransfer(computation, arguments, in ExecuteAndTransfer()
124 StatusOr<Literal> ClientLibraryTestBase::ExecuteAndTransfer( in ExecuteAndTransfer() function in xla::ClientLibraryTestBase
129 return ExecuteAndTransfer(computation, arguments, shape_with_output_layout); in ExecuteAndTransfer()
141 return ref_client_->ExecuteAndTransfer(computation, arguments, in ExecuteAndTransferReference()
154 client_->ExecuteAndTransfer(computation, arguments, &execution_options_); in ExecuteToString()
191 TF_ASSIGN_OR_RETURN(auto actual, ExecuteAndTransfer(computation, arguments)); in ComputeAndCompareLiteralWithAllOutputLayouts()
202 ExecuteAndTransfer(computation, arguments, &layout)); in ComputeAndCompareLiteralWithAllOutputLayouts()
259 ExecuteAndTransfer(computation, in ComputeAndCompareLiteralWithAllInputLayouts()
293 return ExecuteAndTransfer(computation, arguments, shape_with_layout); in ComputeAndTransfer()
[all …]
Dcustom_call_test.cc92 Literal result = ExecuteAndTransfer(std::move(module), {}); in XLA_TEST_F()
113 Literal result = ExecuteAndTransfer(std::move(module), {}); in XLA_TEST_F()
136 Literal result = ExecuteAndTransfer(std::move(module), {}); in XLA_TEST_F()
159 Literal result = ExecuteAndTransfer(std::move(module), {&argument}); in XLA_TEST_F()
186 Literal result = ExecuteAndTransfer(std::move(module), {&argument}); in XLA_TEST_F()
206 Literal result = ExecuteAndTransfer(std::move(module), {&arg0, &arg1}); in XLA_TEST_F()
Dcopy_test.cc51 Literal result = ExecuteAndTransfer(std::move(module), {}); in TestCopyOp()
104 Literal result = ExecuteAndTransfer(std::move(module), {&literal}); in XLA_TEST_F()
124 Literal result = ExecuteAndTransfer(std::move(module), {}); in XLA_TEST_F()
150 Literal result = ExecuteAndTransfer(std::move(module), {}); in XLA_TEST_F()
183 Literal result = ExecuteAndTransfer(std::move(module), {}); in TestCopyConstantLayout021()
217 Literal result = ExecuteAndTransfer(std::move(module), {}); in TestCopyConstantLayoutR4()
253 auto actual = ExecuteAndTransfer(&builder, {input_data.get()}, &out_shape) in XLA_TEST_F()
Dall_reduce_test.cc49 EXPECT_EQ(literal, ExecuteAndTransfer(std::move(module), {&literal})); in XLA_TEST_F()
72 ExecuteAndTransfer(std::move(module), {&literal0, &literal1})); in XLA_TEST_F()
98 ExecuteAndTransfer(std::move(module), {&literal0})); in XLA_TEST_F()
Dreplay_test.cc64 ->ExecuteAndTransfer(replayed, /*arguments=*/{}, &execution_options_) in TEST_F()
103 ->ExecuteAndTransfer(replayed, in XLA_TEST_F()
144 ->ExecuteAndTransfer(replayed, /*arguments=*/{}, &execution_options_) in TEST_F()
Doutfeed_in_nested_computation_test.cc77 comp_result = local_client_->ExecuteAndTransfer(computation, {}) in XLA_TEST_F()
152 comp_result = local_client_->ExecuteAndTransfer(computation, {}) in XLA_TEST_F()
Dcompilation_cache_test.cc47 ->ExecuteAndTransfer(computation, arguments, in ExecuteComputationR0F32()
Dclient_test.cc93 client_->ExecuteAndTransfer(computation, {}, &execution_options)); in XLA_TEST_F()
Dmultioutput_fusion_test.cc119 ExecuteAndTransfer(std::move(hlo_module), {&literal_r0, &arg1}); in RunTest2D()
182 auto actual = ExecuteAndTransfer(std::move(hlo_module), {&input0, &input1}); in RunTest1D()
Dconstants_test.cc168 Literal result = ExecuteAndTransfer(&builder, {}).ConsumeValueOrDie(); in TEST_F()
Dhlo_test_base.h173 Literal ExecuteAndTransfer(std::unique_ptr<HloModule> module,
Dclient_library_test_base.h98 StatusOr<Literal> ExecuteAndTransfer(
102 StatusOr<Literal> ExecuteAndTransfer(
Dscatter_test.cc162 auto actual = ExecuteAndTransfer(std::move(module), {&permutation}); in XLA_TEST_F()
Dscalar_computations_test.cc385 ->ExecuteAndTransfer(div_computation, in XLA_TEST_F()
427 ->ExecuteAndTransfer(rem_computation, in XLA_TEST_F()
Dreshape_test.cc625 ->ExecuteAndTransfer(computation, {input.get()}, &execution_options) in XLA_TEST_P()
775 ->ExecuteAndTransfer(computation, {input_data.get()}, in XLA_TEST_P()
/external/tensorflow/tensorflow/compiler/xla/service/
Dmap_inliner_test.cc72 auto result = ExecuteAndTransfer(hlo_module->Clone(), {}); in TEST_F()
106 auto result = ExecuteAndTransfer(hlo_module->Clone(), {}); in TEST_F()
144 auto result = ExecuteAndTransfer(hlo_module->Clone(), {}); in TEST_F()
175 auto result = ExecuteAndTransfer(hlo_module->Clone(), {}); in TEST_F()
Dhlo_cse_test.cc74 auto result = ExecuteAndTransfer(module->Clone(), {}); in TEST_F()
106 auto result = ExecuteAndTransfer(module->Clone(), {}); in TEST_F()
136 auto result = ExecuteAndTransfer(module->Clone(), {}); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_fusion_test.cc85 auto result = ExecuteAndTransfer(module->Clone(), {}); in TEST_F()
132 auto result = ExecuteAndTransfer(module->Clone(), {}); in TEST_F()
218 auto result = ExecuteAndTransfer(module->Clone(), {}); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dsample_harness.cc61 xla::StatusOr<xla::Literal> result = client->ExecuteAndTransfer( in main()
/external/tensorflow/tensorflow/compiler/xla/rpc/
Dgrpc_client_test.cc100 TF_ASSERT_OK_AND_ASSIGN(auto result_literal, client_->ExecuteAndTransfer( in TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_test.cc91 client->ExecuteAndTransfer(computation, {x_global.get(), y_global.get()}); in TEST()
/external/tensorflow/tensorflow/compiler/xla/client/
Dclient.h170 StatusOr<Literal> ExecuteAndTransfer(

12