Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/tests/
Dmap_test.cc175 client_->TransferToServer(param0_literal).value(); in TEST_F()
189 client_->TransferToServer(param0_literal).value(); in XLA_TEST_F()
204 client_->TransferToServer(param0_literal).value(); in TEST_F()
218 client_->TransferToServer(param0_literal).value(); in TEST_F()
231 client_->TransferToServer(param0_literal).value(); in TEST_F()
245 client_->TransferToServer(param0_literal).value(); in TEST_F()
261 client_->TransferToServer(param0_literal).value(); in XLA_TEST_F()
278 client_->TransferToServer(param0_literal).value(); in TEST_F()
294 client_->TransferToServer(param0_literal).value(); in TEST_F()
349 client_->TransferToServer(param0_literal).value(); in TEST_F()
[all …]
Dparams_test.cc46 client_->TransferToServer(param0_literal).value(); in XLA_TEST_F()
58 client_->TransferToServer(param0_literal).value(); in XLA_TEST_F()
70 client_->TransferToServer(param0_literal).value(); in XLA_TEST_F()
83 client_->TransferToServer(param0_literal).value(); in XLA_TEST_F()
97 client_->TransferToServer(param0_literal).value(); in XLA_TEST_F()
110 client_->TransferToServer(param0_literal).value(); in XLA_TEST_F()
125 client_->TransferToServer(literal0).value(); in XLA_TEST_F()
130 client_->TransferToServer(literal1).value(); in XLA_TEST_F()
155 std::unique_ptr<GlobalData> data = client_->TransferToServer(literal).value(); in XLA_TEST_F()
169 client_->TransferToServer(literal0).value(); in XLA_TEST_F()
[all …]
Dconvolution_test_1d.cc122 auto input_literal = client_->TransferToServer(input_r3).value(); in TestImpl()
123 auto filter_literal = client_->TransferToServer(filter_r3).value(); in TestImpl()
219 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(input)) in XLA_TEST_F()
222 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(filter)) in XLA_TEST_F()
254 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(input)) in RunTest()
257 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(filter)) in RunTest()
289 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(input)) in XLA_TEST_F()
292 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(filter)) in XLA_TEST_F()
320 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(input)) in XLA_TEST_F()
323 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(filter)) in XLA_TEST_F()
[all …]
Dcheck_execution_arity_test.cc46 auto param0_data = client_->TransferToServer(param_literal).value(); in TEST_F()
47 auto param1_data = client_->TransferToServer(param_literal).value(); in TEST_F()
88 auto f32_data = client_->TransferToServer(f32_literal).value(); in XLA_TEST_F()
90 auto f32_4_data = client_->TransferToServer(f32_4_literal).value(); in XLA_TEST_F()
92 auto u8_4_data = client_->TransferToServer(u8_4_literal).value(); in XLA_TEST_F()
Dcompilation_cache_test.cc90 client_->TransferToServer(LiteralUtil::CreateR0<float>(42.0f)).value(); in XLA_TEST_F()
92 client_->TransferToServer(LiteralUtil::CreateR0<float>(123.0f)).value(); in XLA_TEST_F()
94 client_->TransferToServer(LiteralUtil::CreateR0<float>(456.0f)).value(); in XLA_TEST_F()
143 auto rowmaj_handle = client_->TransferToServer(rowmaj_array).value(); in XLA_TEST_F()
147 auto colmaj_handle = client_->TransferToServer(colmaj_array).value(); in XLA_TEST_F()
Dconvolution_test.cc304 auto input_literal = client_->TransferToServer(input_r5).value(); in XLA_TEST_F()
305 auto filter_literal = client_->TransferToServer(filter_r5).value(); in XLA_TEST_F()
365 auto input_literal = client_->TransferToServer(input_r4).value(); in RunTest()
366 auto filter_literal = client_->TransferToServer(filter_r4).value(); in RunTest()
427 auto input_literal = client_->TransferToServer(input_r4).value(); in RunTest()
428 auto filter_literal = client_->TransferToServer(filter_r4).value(); in RunTest()
493 auto input_literal = client_->TransferToServer(input_r4).value(); in RunTest()
494 auto filter_literal = client_->TransferToServer(filter_r4).value(); in RunTest()
556 auto input_literal = client_->TransferToServer(input_r4).value(); in RunTest()
557 auto filter_literal = client_->TransferToServer(filter_r4).value(); in RunTest()
[all …]
Ddot_operation_test.cc206 ->TransferToServer(LiteralUtil::CreateR2FromArray2D<T>( in XLA_TYPED_TEST()
210 ->TransferToServer(LiteralUtil::CreateR2FromArray2D<T>( in XLA_TYPED_TEST()
229 ->TransferToServer(LiteralUtil::CreateFromArrayWithLayout<T>( in TestImpl()
236 ->TransferToServer(LiteralUtil::CreateFromArrayWithLayout<T>( in TestImpl()
328 client_->TransferToServer(dot_lhs_lit).value(); in TestImpl()
337 client_->TransferToServer(dot_rhs_lit).value(); in TestImpl()
348 addend_handle = client_->TransferToServer(addend_lit).value(); in TestImpl()
514 ->TransferToServer(LiteralUtil::CreateFromArrayWithLayout<T>( in TestImpl()
521 ->TransferToServer(LiteralUtil::CreateFromArrayWithLayout<T>( in TestImpl()
548 ->TransferToServer(LiteralUtil::CreateR2WithLayout<complex64>( in XLA_TEST_F()
[all …]
Dbatch_normalization_test.cc594 client_->TransferToServer(input_literal).value(); in XLA_TEST_P()
596 client_->TransferToServer(scale_literal).value(); in XLA_TEST_P()
598 client_->TransferToServer(offset_literal).value(); in XLA_TEST_P()
695 client_->TransferToServer(input_literal).value(); in XLA_TEST_P()
697 client_->TransferToServer(scale_literal).value(); in XLA_TEST_P()
699 client_->TransferToServer(offset_literal).value(); in XLA_TEST_P()
701 client_->TransferToServer(mean_literal).value(); in XLA_TEST_P()
703 client_->TransferToServer(var_literal).value(); in XLA_TEST_P()
880 client_->TransferToServer(input_literal).value(); in XLA_TEST_P()
882 client_->TransferToServer(scale_literal).value(); in XLA_TEST_P()
[all …]
Dconvert_test.cc215 client_->TransferToServer(arg_literal).value(); in XLA_TEST_F()
234 client_->TransferToServer(arg_literal).value(); in XLA_TEST_F()
252 client_->TransferToServer(arg_literal).value(); in XLA_TEST_F()
269 client_->TransferToServer(arg_literal).value(); in XLA_TEST_F()
286 client_->TransferToServer(arg_literal).value(); in XLA_TEST_F()
323 client_->TransferToServer(arg_literal).value(); in XLA_TEST_F()
457 client_->TransferToServer(LiteralUtil::CreateR1<half>(input))); in XLA_TEST_F()
477 client_->TransferToServer(LiteralUtil::CreateR1<float>(input))); in XLA_TEST_F()
Darray_elementwise_ops_test.cc268 client_->TransferToServer(lhs_literal).value(); in XLA_TEST_F()
282 client_->TransferToServer(rhs_literal).value(); in XLA_TEST_F()
308 client_->TransferToServer(lhs_literal).value(); in XLA_TEST_F()
321 client_->TransferToServer(rhs_literal).value(); in XLA_TEST_F()
363 client_->TransferToServer(a_literal).value(); in TEST_P()
369 client_->TransferToServer(b_literal).value(); in TEST_P()
423 auto a_data = client_->TransferToServer(a_literal).value(); in XLA_TEST_F()
424 auto b_data = client_->TransferToServer(b_literal).value(); in XLA_TEST_F()
1600 client_->TransferToServer(param_literal).value(); in XLA_TEST_F()
1629 client_->TransferToServer(literal0).value(); in XLA_TEST_F()
[all …]
Dconcat_test.cc520 auto a_data = client_->TransferToServer(a_literal).value(); in XLA_TEST_F()
811 auto x_data = client_->TransferToServer(x_literal).value(); in XLA_TEST_F()
812 auto y_data = client_->TransferToServer(y_literal).value(); in XLA_TEST_F()
834 auto x_data = client_->TransferToServer(x_literal).value(); in XLA_TEST_F()
835 auto y_data = client_->TransferToServer(y_literal).value(); in XLA_TEST_F()
836 auto z_data = client_->TransferToServer(z_literal).value(); in XLA_TEST_F()
862 auto x_data = client_->TransferToServer(x_literal).value(); in XLA_TEST_F()
863 auto y_data = client_->TransferToServer(y_literal).value(); in XLA_TEST_F()
864 auto z_data = client_->TransferToServer(z_literal).value(); in XLA_TEST_F()
Dreduce_test.cc114 client_->TransferToServer(input_literal).value(); in RunR1ToR0Test()
142 client_->TransferToServer(input_literal).value(); in RunR1ToR0PredTest()
185 client_->TransferToServer(input_literal).value(); in RunR2ToR1PredTest()
219 client_->TransferToServer(input_literal).value(); in RunR2ToR0Test()
247 client_->TransferToServer(input_literal).value(); in RunR2ToR1Test()
305 client_->TransferToServer(input_literal).value(); in RunVectorizedReduceTestForType()
458 client_->TransferToServer(input_literal).value(); in XLA_TEST_F()
490 client_->TransferToServer(input_literal).value(); in XLA_TEST_F()
537 client_->TransferToServer(input_literal).value(); in XLA_TEST_F()
830 client_->TransferToServer(input_literal).value(); in XLA_TEST_P()
[all …]
Dscalar_computations_test.cc165 client_->TransferToServer(a_literal).value(); in XLA_TEST_F()
242 client_->TransferToServer(a_literal).value(); in XLA_TEST_F()
244 client_->TransferToServer(b_literal).value(); in XLA_TEST_F()
246 client_->TransferToServer(c_literal).value(); in XLA_TEST_F()
389 client_->TransferToServer(dividend_literal)); in XLA_TEST_F()
391 client_->TransferToServer(divisor_literal)); in XLA_TEST_F()
431 client_->TransferToServer(dividend_literal)); in XLA_TEST_F()
433 client_->TransferToServer(divisor_literal)); in XLA_TEST_F()
454 TF_ASSERT_OK_AND_ASSIGN(auto input_data, client_->TransferToServer(literal)); in XLA_TEST_F()
Dclient_library_test_base.cc246 client_->TransferToServer(literal_relayout)); in ComputeAndCompareLiteralWithAllInputLayouts()
286 client_->TransferToServer(MaybeConvertLiteralToBfloat16(argument))); in ComputeAndTransfer()
310 client_->TransferToServer(MaybeConvertLiteralToBfloat16(argument))); in ComputeAndCompareLiteralWithStatus()
372 client_->TransferToServer(MaybeConvertLiteralToBfloat16(argument))); in ComputeAndCompareLiteralWithStatus()
501 TF_ASSIGN_OR_RETURN(auto data, client_->TransferToServer(arg.Clone())); in ComputeValueAndReference()
502 TF_ASSIGN_OR_RETURN(auto ref_data, ref_client_->TransferToServer(arg)); in ComputeValueAndReference()
647 client_->TransferToServer(param_literal, device_handle)); in CreateParameterAndTransferLiteral()
Dcholesky_test.cc256 TF_ASSERT_OK_AND_ASSIGN(auto input_data, client_->TransferToServer(literal)); in XLA_TEST_P()
302 client_->TransferToServer(literal_real)); in XLA_TEST_P()
304 client_->TransferToServer(literal_imag)); in XLA_TEST_P()
Dreplay_test.cc93 client_->TransferToServer(LiteralUtil::CreateR0<int32_t>(2)).value(); in XLA_TEST_F()
95 client_->TransferToServer(LiteralUtil::CreateR0<int32_t>(3)).value(); in XLA_TEST_F()
Dtuple_test.cc285 ->TransferToServer(LiteralUtil::MakeTupleFromSlices({ in XLA_TEST_F()
321 ->TransferToServer(LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F()
332 ->TransferToServer( in XLA_TEST_F()
Dclient_library_test_base.h612 std::unique_ptr<GlobalData> data = client_->TransferToServer(literal).value(); in CreateR0Parameter()
625 std::unique_ptr<GlobalData> data = client_->TransferToServer(literal).value(); in CreateR1Parameter()
638 std::unique_ptr<GlobalData> data = client_->TransferToServer(literal).value(); in CreateR2Parameter()
651 std::unique_ptr<GlobalData> data = client_->TransferToServer(literal).value(); in CreateR3Parameter()
664 std::unique_ptr<GlobalData> data = client_->TransferToServer(literal).value(); in CreateR4Parameter()
677 std::unique_ptr<GlobalData> data = client_->TransferToServer(literal).value(); in CreateParameter()
Dexecution_profile_test.cc33 client_->TransferToServer( in XLA_TEST_F()
Dmatrix_ops_simple_test.cc201 client_->TransferToServer(LiteralUtil::CreateR2FromArray2DWithLayout<T>( in TestImpl()
205 client_->TransferToServer(LiteralUtil::CreateR2FromArray2DWithLayout<T>( in TestImpl()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_test.cc80 auto x_global_or = client->TransferToServer(x_literal); in TEST()
81 auto y_global_or = client->TransferToServer(y_literal); in TEST()
118 auto x_global_or = client->TransferToServer(x_literal); in TEST()
119 auto y_global_or = client->TransferToServer(y_literal); in TEST()
120 auto unused_global_or = client->TransferToServer(y_literal); in TEST()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dbef_executable_test.cc36 client_->TransferToServer( in XLA_TEST_F()
40 client_->TransferToServer( in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dsample_harness.cc44 client->TransferToServer(param0_literal).value(); in main()
49 client->TransferToServer(param1_literal).value(); in main()
/external/tensorflow/tensorflow/compiler/xla/rpc/
Dgrpc_service.cc96 ::grpc::Status GRPCService::TransferToServer(::grpc::ServerContext* context, in TransferToServer() function in xla::GRPCService
100 return service_->TransferToServer(arg, result); in TransferToServer()
Dgrpc_stub.cc37 Status GRPCStub::TransferToServer(const TransferToServerRequest* request, in TransferToServer() function in xla::GRPCStub
40 return grpc_stub_->TransferToServer(context, *request, response); in TransferToServer()

12