Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/tests/
Dprng_test.cc60 ExecuteAndTransfer(&builder, /*arguments=*/{}).ConsumeValueOrDie(); in UniformTest()
158 ExecuteAndTransfer(&builder, /*arguments=*/{}).ConsumeValueOrDie(); in UniformChiSquared()
219 auto actual, client_->ExecuteAndTransfer( in XLA_TEST_F()
255 result1, client_->ExecuteAndTransfer(computation, /*arguments=*/{}, in XLA_TEST_F()
263 result2, client_->ExecuteAndTransfer(computation, /*arguments=*/{}, in XLA_TEST_F()
266 result3, client_->ExecuteAndTransfer(computation, /*arguments=*/{}, in XLA_TEST_F()
276 result4, client_->ExecuteAndTransfer(computation, /*arguments=*/{}, in XLA_TEST_F()
279 result5, client_->ExecuteAndTransfer(computation, /*arguments=*/{}, in XLA_TEST_F()
282 result6, client_->ExecuteAndTransfer(computation, /*arguments=*/{}, in XLA_TEST_F()
316 result_tuple, client_->ExecuteAndTransfer(computation, /*arguments=*/{}, in XLA_TEST_F()
[all …]
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 …]
Dcpu_gpu_fusion_test.cc130 auto actual = ExecuteAndTransfer(std::move(hlo_module), {}); in TestElementwise2D()
241 ExecuteAndTransfer(std::move(hlo_module), {}), ErrorSpec(1e-4))); in XLA_TEST_F()
267 ExecuteAndTransfer(std::move(hlo_module), {}), ErrorSpec(1e-4))); in XLA_TEST_F()
297 auto result = ExecuteAndTransfer(std::move(hlo_module), {}); in XLA_TEST_F()
327 ExecuteAndTransfer(std::move(hlo_module), {}), ErrorSpec(1e-4))); in XLA_TEST_F()
342 ExecuteAndTransfer(std::move(hlo_module), {}))); in XLA_TEST_F()
357 ExecuteAndTransfer(std::move(hlo_module), {}))); in XLA_TEST_F()
372 ExecuteAndTransfer(std::move(hlo_module), {}))); in XLA_TEST_F()
387 ExecuteAndTransfer(std::move(hlo_module), {}))); in XLA_TEST_F()
402 ExecuteAndTransfer(std::move(hlo_module), {}))); in XLA_TEST_F()
[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()
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 …]
Dall_reduce_test.cc49 EXPECT_EQ(literal, ExecuteAndTransfer(std::move(module), {&literal})); in XLA_TEST_F()
73 ExecuteAndTransfer(std::move(module), {&literal0, &literal1})); in XLA_TEST_F()
100 ExecuteAndTransfer(std::move(module), {&literal0})); in XLA_TEST_F()
141 ExecuteAndTransfer(std::move(module), {&literal_in})); in XLA_TEST_F()
183 ExecuteAndTransfer(std::move(module), {&literal_in})); 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()
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()
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()
Dcompilation_cache_test.cc47 ->ExecuteAndTransfer(computation, arguments, in ExecuteComputationR0F32()
Dclient_test.cc93 client_->ExecuteAndTransfer(computation, {}, &execution_options)); in XLA_TEST_F()
Dconstants_test.cc168 Literal result = ExecuteAndTransfer(&builder, {}).ConsumeValueOrDie(); in TEST_F()
Dhlo_test_base.h154 Literal ExecuteAndTransfer(std::unique_ptr<HloModule> module,
Dscatter_test.cc161 auto actual = ExecuteAndTransfer(std::move(module), {&permutation}); in XLA_TEST_F()
Dclient_library_test_base.h100 StatusOr<Literal> ExecuteAndTransfer(
104 StatusOr<Literal> ExecuteAndTransfer(
/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()
Ddynamic_padder_test.cc393 return ExecuteAndTransfer(std::move(module), arguments); in PadAndExecute()
433 ExecuteAndTransfer(std::move(module_not_padded), in XLA_TEST_F()
597 Literal not_padded = ExecuteAndTransfer(std::move(module_not_padded), in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcustom_call_test.cc93 TF_ASSERT_OK_AND_ASSIGN(auto result, ExecuteAndTransfer(&b, {})); in TEST_F()
162 TF_ASSERT_OK_AND_ASSIGN(auto result, ExecuteAndTransfer(&b, {})); in TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_test.cc91 client->ExecuteAndTransfer(computation, {x_global.get(), y_global.get()}); in TEST()
132 auto result_or = client->ExecuteAndTransfer( in TEST()
/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.cc106 TF_ASSERT_OK_AND_ASSIGN(auto result_literal, client_->ExecuteAndTransfer( in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/client/
Dclient.h170 StatusOr<Literal> ExecuteAndTransfer(

12