Home
last modified time | relevance | path

Searched refs:TransferToServer (Results 1 – 25 of 41) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/tests/
Ddot_operation_test.cc160 ->TransferToServer(*Literal::CreateR2<float>( in XLA_TEST_F()
164 ->TransferToServer(*Literal::CreateR2<float>( in XLA_TEST_F()
178 ->TransferToServer(*Literal::CreateR2WithLayout<Element>( in TestSquareMatrixDot()
184 ->TransferToServer(*Literal::CreateR2WithLayout<Element>( in TestSquareMatrixDot()
239 client_->TransferToServer(*dot_lhs_lit).ConsumeValueOrDie(); in XLA_TEST_P()
247 client_->TransferToServer(*dot_rhs_lit).ConsumeValueOrDie(); in XLA_TEST_P()
258 addend_handle = client_->TransferToServer(*addend_lit).ConsumeValueOrDie(); in XLA_TEST_P()
387 ->TransferToServer(*Literal::CreateR2WithLayout<Element>( in TestNonsquareMatrixDot()
393 ->TransferToServer(*Literal::CreateR2WithLayout<Element>( in TestNonsquareMatrixDot()
449 ->TransferToServer(*Literal::CreateR2WithLayout<complex64>( in XLA_TEST_F()
[all …]
Dmap_test.cc173 client_->TransferToServer(*param0_literal).ConsumeValueOrDie(); in TEST_F()
187 client_->TransferToServer(*param0_literal).ConsumeValueOrDie(); in XLA_TEST_F()
202 client_->TransferToServer(*param0_literal).ConsumeValueOrDie(); in TEST_F()
216 client_->TransferToServer(*param0_literal).ConsumeValueOrDie(); in TEST_F()
229 client_->TransferToServer(*param0_literal).ConsumeValueOrDie(); in TEST_F()
243 client_->TransferToServer(*param0_literal).ConsumeValueOrDie(); in TEST_F()
259 client_->TransferToServer(*param0_literal).ConsumeValueOrDie(); in XLA_TEST_F()
276 client_->TransferToServer(*param0_literal).ConsumeValueOrDie(); in TEST_F()
292 client_->TransferToServer(*param0_literal).ConsumeValueOrDie(); in TEST_F()
388 client_->TransferToServer(*param0_literal).ConsumeValueOrDie(); in TEST_F()
[all …]
Dparams_test.cc47 client_->TransferToServer(*param0_literal).ConsumeValueOrDie(); in XLA_TEST_F()
59 client_->TransferToServer(*param0_literal).ConsumeValueOrDie(); in XLA_TEST_F()
72 client_->TransferToServer(*param0_literal).ConsumeValueOrDie(); in XLA_TEST_F()
85 client_->TransferToServer(*param0_literal).ConsumeValueOrDie(); in XLA_TEST_F()
98 client_->TransferToServer(*param0_literal).ConsumeValueOrDie(); in XLA_TEST_F()
111 client_->TransferToServer(*param0_literal).ConsumeValueOrDie(); in XLA_TEST_F()
126 client_->TransferToServer(*literal0).ConsumeValueOrDie(); in XLA_TEST_F()
131 client_->TransferToServer(*literal1).ConsumeValueOrDie(); in XLA_TEST_F()
157 client_->TransferToServer(*literal).ConsumeValueOrDie(); in XLA_TEST_F()
175 client_->TransferToServer(*literal0).ConsumeValueOrDie(); in XLA_TEST_F()
[all …]
Dconvolution_test.cc267 client_->TransferToServer(*Literal::CreateR3FromArray3D(input)) in XLA_TEST_F()
270 client_->TransferToServer(*Literal::CreateR3FromArray3D(filter)) in XLA_TEST_F()
302 client_->TransferToServer(*Literal::CreateR3FromArray3D(input)) in RunTest()
305 client_->TransferToServer(*Literal::CreateR3FromArray3D(filter)) in RunTest()
337 client_->TransferToServer(*Literal::CreateR3FromArray3D(input)) in XLA_TEST_F()
340 client_->TransferToServer(*Literal::CreateR3FromArray3D(filter)) in XLA_TEST_F()
368 client_->TransferToServer(*Literal::CreateR3FromArray3D(input)) in XLA_TEST_F()
371 client_->TransferToServer(*Literal::CreateR3FromArray3D(filter)) in XLA_TEST_F()
404 client_->TransferToServer(*Literal::CreateR3FromArray3D(input)) in RunTest()
407 client_->TransferToServer(*Literal::CreateR3FromArray3D(filter)) in RunTest()
[all …]
Dcheck_execution_arity_test.cc46 client_->TransferToServer(*param_literal).ConsumeValueOrDie(); in TEST_F()
48 client_->TransferToServer(*param_literal).ConsumeValueOrDie(); in TEST_F()
89 auto f32_data = client_->TransferToServer(*f32_literal).ConsumeValueOrDie(); in XLA_TEST_F()
92 client_->TransferToServer(*f32_4_literal).ConsumeValueOrDie(); in XLA_TEST_F()
94 auto u8_4_data = client_->TransferToServer(*u8_4_literal).ConsumeValueOrDie(); in XLA_TEST_F()
Dcompilation_cache_test.cc89 client_->TransferToServer(*Literal::CreateR0<float>(42.0f)) in XLA_TEST_F()
92 client_->TransferToServer(*Literal::CreateR0<float>(123.0f)) in XLA_TEST_F()
95 client_->TransferToServer(*Literal::CreateR0<float>(456.0f)) in XLA_TEST_F()
144 client_->TransferToServer(*rowmaj_array).ConsumeValueOrDie(); in XLA_TEST_F()
149 client_->TransferToServer(*colmaj_array).ConsumeValueOrDie(); in XLA_TEST_F()
Dreduce_precision_test.cc235 client_->TransferToServer(*a_literal).ConsumeValueOrDie(); in XLA_TEST_P()
259 client_->TransferToServer(*a_literal).ConsumeValueOrDie(); in XLA_TEST_F()
288 client_->TransferToServer(*a_literal).ConsumeValueOrDie(); in XLA_TEST_F()
314 client_->TransferToServer(*a_literal).ConsumeValueOrDie(); in XLA_TEST_F()
338 client_->TransferToServer(*a_literal).ConsumeValueOrDie(); in XLA_TEST_F()
363 client_->TransferToServer(*a_literal).ConsumeValueOrDie(); in XLA_TEST_F()
Dbatch_normalization_test.cc521 client_->TransferToServer(*input_literal).ConsumeValueOrDie(); in XLA_TEST_P()
523 client_->TransferToServer(*scale_literal).ConsumeValueOrDie(); in XLA_TEST_P()
525 client_->TransferToServer(*offset_literal).ConsumeValueOrDie(); in XLA_TEST_P()
622 client_->TransferToServer(*input_literal).ConsumeValueOrDie(); in XLA_TEST_P()
624 client_->TransferToServer(*scale_literal).ConsumeValueOrDie(); in XLA_TEST_P()
626 client_->TransferToServer(*offset_literal).ConsumeValueOrDie(); in XLA_TEST_P()
628 client_->TransferToServer(*mean_literal).ConsumeValueOrDie(); in XLA_TEST_P()
630 client_->TransferToServer(*var_literal).ConsumeValueOrDie(); in XLA_TEST_P()
807 client_->TransferToServer(*input_literal).ConsumeValueOrDie(); in XLA_TEST_P()
809 client_->TransferToServer(*scale_literal).ConsumeValueOrDie(); in XLA_TEST_P()
[all …]
Dconcat_test.cc521 auto x_data = client_->TransferToServer(*x_literal).ConsumeValueOrDie(); in XLA_TEST_F()
522 auto y_data = client_->TransferToServer(*y_literal).ConsumeValueOrDie(); in XLA_TEST_F()
544 auto x_data = client_->TransferToServer(*x_literal).ConsumeValueOrDie(); in XLA_TEST_F()
545 auto y_data = client_->TransferToServer(*y_literal).ConsumeValueOrDie(); in XLA_TEST_F()
546 auto z_data = client_->TransferToServer(*z_literal).ConsumeValueOrDie(); in XLA_TEST_F()
572 auto x_data = client_->TransferToServer(*x_literal).ConsumeValueOrDie(); in XLA_TEST_F()
573 auto y_data = client_->TransferToServer(*y_literal).ConsumeValueOrDie(); in XLA_TEST_F()
574 auto z_data = client_->TransferToServer(*z_literal).ConsumeValueOrDie(); in XLA_TEST_F()
Darray_elementwise_ops_test.cc201 client_->TransferToServer(*a_literal).ConsumeValueOrDie(); in TEST_P()
207 client_->TransferToServer(*b_literal).ConsumeValueOrDie(); in TEST_P()
1227 client_->TransferToServer(*param_literal).ConsumeValueOrDie(); in XLA_TEST_F()
1255 client_->TransferToServer(*literal0).ConsumeValueOrDie(); in XLA_TEST_F()
1258 client_->TransferToServer(*literal1).ConsumeValueOrDie(); in XLA_TEST_F()
1280 client_->TransferToServer(*literal0).ConsumeValueOrDie(); in XLA_TEST_F()
1283 client_->TransferToServer(*literal1).ConsumeValueOrDie(); in XLA_TEST_F()
1305 client_->TransferToServer(*literal0).ConsumeValueOrDie(); in XLA_TEST_F()
1308 client_->TransferToServer(*literal1).ConsumeValueOrDie(); in XLA_TEST_F()
1330 client_->TransferToServer(*literal0).ConsumeValueOrDie(); in XLA_TEST_F()
[all …]
Dreduce_test.cc101 client_->TransferToServer(*input_literal).ConsumeValueOrDie(); in RunR1ToR0Test()
133 client_->TransferToServer(*input_literal).ConsumeValueOrDie(); in RunR1ToR0PredTest()
177 client_->TransferToServer(*input_literal).ConsumeValueOrDie(); in RunR2ToR1PredTest()
211 client_->TransferToServer(*input_literal).ConsumeValueOrDie(); in RunR2ToR0Test()
239 client_->TransferToServer(*input_literal).ConsumeValueOrDie(); in RunR2ToR1Test()
297 client_->TransferToServer(*input_literal).ConsumeValueOrDie(); in RunVectorizedReduceTestForType()
451 client_->TransferToServer(*input_literal).ConsumeValueOrDie(); in XLA_TEST_F()
483 client_->TransferToServer(*input_literal).ConsumeValueOrDie(); in XLA_TEST_F()
534 client_->TransferToServer(*input_literal).ConsumeValueOrDie(); in XLA_TEST_F()
821 client_->TransferToServer(*input_literal).ConsumeValueOrDie(); in XLA_TEST_P()
[all …]
Dmatrix_ops_simple_test.cc183 client_->TransferToServer(*Literal::CreateR2FromArray2DWithLayout<float>( in TEST_P()
187 client_->TransferToServer(*Literal::CreateR2FromArray2DWithLayout<float>( in TEST_P()
249 client_->TransferToServer( in TEST_P()
254 client_->TransferToServer( in TEST_P()
Dscalar_computations_test.cc169 client_->TransferToServer(*a_literal).ConsumeValueOrDie(); in XLA_TEST_F()
238 client_->TransferToServer(*a_literal).ConsumeValueOrDie(); in XLA_TEST_F()
240 client_->TransferToServer(*b_literal).ConsumeValueOrDie(); in XLA_TEST_F()
242 client_->TransferToServer(*c_literal).ConsumeValueOrDie(); in XLA_TEST_F()
385 client_->TransferToServer(*dividend_literal)); in XLA_TEST_F()
387 client_->TransferToServer(*divisor_literal)); in XLA_TEST_F()
426 client_->TransferToServer(*dividend_literal)); in XLA_TEST_F()
428 client_->TransferToServer(*divisor_literal)); in XLA_TEST_F()
448 TF_ASSERT_OK_AND_ASSIGN(auto input_data, client_->TransferToServer(*literal)); in XLA_TEST_F()
900 client_->TransferToServer(zero_literal).ConsumeValueOrDie(); in XLA_TEST_F()
Dreplay_test.cc94 client_->TransferToServer(*Literal::CreateR0<int32>(2)) in XLA_TEST_F()
97 client_->TransferToServer(*Literal::CreateR0<int32>(3)) in XLA_TEST_F()
Dexecution_profile_test.cc33 client_->TransferToServer( in XLA_TEST_F()
Dclient_library_test_base.h556 client_->TransferToServer(*literal).ConsumeValueOrDie(); in CreateR0Parameter()
571 client_->TransferToServer(*literal).ConsumeValueOrDie(); in CreateR1Parameter()
586 client_->TransferToServer(*literal).ConsumeValueOrDie(); in CreateR2Parameter()
601 client_->TransferToServer(*literal).ConsumeValueOrDie(); in CreateR3Parameter()
Dconvolution_dimension_numbers_test.cc76 client_->TransferToServer(*Literal::CreateR4FromArray4D(*weight_array)) in XLA_TEST_F()
Dtuple_test.cc445 ->TransferToServer(*Literal::MakeTuple({ in XLA_TEST_F()
483 ->TransferToServer(*Literal::MakeTuple( in XLA_TEST_F()
496 ->TransferToServer(*Literal::CreateR1<complex64>({{1, 2}, {1, -2}})) in XLA_TEST_F()
Dround_trip_transfer_test.cc41 client_->TransferToServer(original).ConsumeValueOrDie(); in RoundTripTest()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dsample_harness.cc43 client->TransferToServer(*param0_literal).ConsumeValueOrDie(); in main()
48 client->TransferToServer(*param1_literal).ConsumeValueOrDie(); in main()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_test.cc78 auto x_global_or = client->TransferToServer(*x_literal); in TEST()
79 auto y_global_or = client->TransferToServer(*y_literal); in TEST()
Dxla_compiler_test.cc213 client_->TransferToServer(*param0_literal).ConsumeValueOrDie(); in TEST_F()
215 client_->TransferToServer(*param1_literal).ConsumeValueOrDie(); in TEST_F()
311 client_->TransferToServer(*param0_literal).ConsumeValueOrDie(); in TEST_F()
345 client_->TransferToServer(*param0_literal).ConsumeValueOrDie(); in TEST_F()
502 client_->TransferToServer(*input).ConsumeValueOrDie(); in TEST_F()
725 client_->TransferToServer(*param0_literal).ConsumeValueOrDie(); in TEST_F()
727 client_->TransferToServer(*param1_literal).ConsumeValueOrDie(); in TEST_F()
790 client_->TransferToServer(*param0_literal).ConsumeValueOrDie(); in TEST_F()
792 client_->TransferToServer(*param1_literal).ConsumeValueOrDie(); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dcompile_only_service.h81 tensorflow::Status TransferToServer( in TransferToServer() function
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dtesting.cc92 return client->TransferToServer(*literal_status.ValueOrDie()).ValueOrDie(); in MakeFakeDataOrDie()
/external/tensorflow/tensorflow/compiler/xla/client/
Dclient.h108 StatusOr<std::unique_ptr<GlobalData>> TransferToServer(

12