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.cc174 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in TEST_F()
188 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in XLA_TEST_F()
203 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in TEST_F()
217 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in TEST_F()
230 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in TEST_F()
244 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in TEST_F()
260 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in XLA_TEST_F()
277 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in TEST_F()
293 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in TEST_F()
348 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()
71 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in XLA_TEST_F()
84 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()
171 client_->TransferToServer(literal0).ConsumeValueOrDie(); in XLA_TEST_F()
[all …]
Dconvolution_test_1d.cc125 client_->TransferToServer(input_r3).ConsumeValueOrDie(); in TestImpl()
127 client_->TransferToServer(filter_r3).ConsumeValueOrDie(); in TestImpl()
223 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(input)) in XLA_TEST_F()
226 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(filter)) in XLA_TEST_F()
258 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(input)) in RunTest()
261 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(filter)) in RunTest()
293 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(input)) in XLA_TEST_F()
296 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(filter)) in XLA_TEST_F()
324 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(input)) in XLA_TEST_F()
327 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(filter)) in XLA_TEST_F()
[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()
Dconvolution_test.cc289 auto input_literal = client_->TransferToServer(input_r5).ConsumeValueOrDie(); in XLA_TEST_F()
291 client_->TransferToServer(filter_r5).ConsumeValueOrDie(); in XLA_TEST_F()
352 client_->TransferToServer(input_r4).ConsumeValueOrDie(); in RunTest()
354 client_->TransferToServer(filter_r4).ConsumeValueOrDie(); in RunTest()
416 client_->TransferToServer(input_r4).ConsumeValueOrDie(); in RunTest()
418 client_->TransferToServer(filter_r4).ConsumeValueOrDie(); in RunTest()
484 client_->TransferToServer(input_r4).ConsumeValueOrDie(); in RunTest()
486 client_->TransferToServer(filter_r4).ConsumeValueOrDie(); in RunTest()
549 client_->TransferToServer(input_r4).ConsumeValueOrDie(); in RunTest()
551 client_->TransferToServer(filter_r4).ConsumeValueOrDie(); in RunTest()
[all …]
Dcompilation_cache_test.cc90 client_->TransferToServer(LiteralUtil::CreateR0<float>(42.0f)) in XLA_TEST_F()
93 client_->TransferToServer(LiteralUtil::CreateR0<float>(123.0f)) in XLA_TEST_F()
96 client_->TransferToServer(LiteralUtil::CreateR0<float>(456.0f)) in XLA_TEST_F()
147 client_->TransferToServer(rowmaj_array).ConsumeValueOrDie(); in XLA_TEST_F()
152 client_->TransferToServer(colmaj_array).ConsumeValueOrDie(); in XLA_TEST_F()
Ddot_operation_test.cc207 ->TransferToServer(LiteralUtil::CreateR2FromArray2D<T>( in XLA_TYPED_TEST()
211 ->TransferToServer(LiteralUtil::CreateR2FromArray2D<T>( in XLA_TYPED_TEST()
230 ->TransferToServer(LiteralUtil::CreateFromArrayWithLayout<T>( in TestImpl()
237 ->TransferToServer(LiteralUtil::CreateFromArrayWithLayout<T>( in TestImpl()
329 client_->TransferToServer(dot_lhs_lit).ConsumeValueOrDie(); in TestImpl()
338 client_->TransferToServer(dot_rhs_lit).ConsumeValueOrDie(); in TestImpl()
349 addend_handle = client_->TransferToServer(addend_lit).ConsumeValueOrDie(); in TestImpl()
515 ->TransferToServer(LiteralUtil::CreateFromArrayWithLayout<T>( in TestImpl()
522 ->TransferToServer(LiteralUtil::CreateFromArrayWithLayout<T>( in TestImpl()
549 ->TransferToServer(LiteralUtil::CreateR2WithLayout<complex64>( in XLA_TEST_F()
[all …]
Dbatch_normalization_test.cc629 client_->TransferToServer(input_literal).ConsumeValueOrDie(); in XLA_TEST_P()
631 client_->TransferToServer(scale_literal).ConsumeValueOrDie(); in XLA_TEST_P()
633 client_->TransferToServer(offset_literal).ConsumeValueOrDie(); in XLA_TEST_P()
730 client_->TransferToServer(input_literal).ConsumeValueOrDie(); in XLA_TEST_P()
732 client_->TransferToServer(scale_literal).ConsumeValueOrDie(); in XLA_TEST_P()
734 client_->TransferToServer(offset_literal).ConsumeValueOrDie(); in XLA_TEST_P()
736 client_->TransferToServer(mean_literal).ConsumeValueOrDie(); in XLA_TEST_P()
738 client_->TransferToServer(var_literal).ConsumeValueOrDie(); in XLA_TEST_P()
915 client_->TransferToServer(input_literal).ConsumeValueOrDie(); in XLA_TEST_P()
917 client_->TransferToServer(scale_literal).ConsumeValueOrDie(); in XLA_TEST_P()
[all …]
Dconvert_test.cc216 client_->TransferToServer(arg_literal).ConsumeValueOrDie(); in XLA_TEST_F()
235 client_->TransferToServer(arg_literal).ConsumeValueOrDie(); in XLA_TEST_F()
253 client_->TransferToServer(arg_literal).ConsumeValueOrDie(); in XLA_TEST_F()
270 client_->TransferToServer(arg_literal).ConsumeValueOrDie(); in XLA_TEST_F()
287 client_->TransferToServer(arg_literal).ConsumeValueOrDie(); in XLA_TEST_F()
324 client_->TransferToServer(arg_literal).ConsumeValueOrDie(); 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.cc264 client_->TransferToServer(lhs_literal).ConsumeValueOrDie(); in XLA_TEST_F()
278 client_->TransferToServer(rhs_literal).ConsumeValueOrDie(); in XLA_TEST_F()
304 client_->TransferToServer(lhs_literal).ConsumeValueOrDie(); in XLA_TEST_F()
317 client_->TransferToServer(rhs_literal).ConsumeValueOrDie(); in XLA_TEST_F()
357 client_->TransferToServer(a_literal).ConsumeValueOrDie(); in TEST_P()
363 client_->TransferToServer(b_literal).ConsumeValueOrDie(); in TEST_P()
416 auto a_data = client_->TransferToServer(a_literal).ConsumeValueOrDie(); in XLA_TEST_F()
417 auto b_data = client_->TransferToServer(b_literal).ConsumeValueOrDie(); in XLA_TEST_F()
1586 client_->TransferToServer(param_literal).ConsumeValueOrDie(); in XLA_TEST_F()
1614 client_->TransferToServer(literal0).ConsumeValueOrDie(); in XLA_TEST_F()
[all …]
Dconcat_test.cc520 auto a_data = client_->TransferToServer(a_literal).ConsumeValueOrDie(); in XLA_TEST_F()
811 auto x_data = client_->TransferToServer(x_literal).ConsumeValueOrDie(); in XLA_TEST_F()
812 auto y_data = client_->TransferToServer(y_literal).ConsumeValueOrDie(); in XLA_TEST_F()
834 auto x_data = client_->TransferToServer(x_literal).ConsumeValueOrDie(); in XLA_TEST_F()
835 auto y_data = client_->TransferToServer(y_literal).ConsumeValueOrDie(); in XLA_TEST_F()
836 auto z_data = client_->TransferToServer(z_literal).ConsumeValueOrDie(); in XLA_TEST_F()
862 auto x_data = client_->TransferToServer(x_literal).ConsumeValueOrDie(); in XLA_TEST_F()
863 auto y_data = client_->TransferToServer(y_literal).ConsumeValueOrDie(); in XLA_TEST_F()
864 auto z_data = client_->TransferToServer(z_literal).ConsumeValueOrDie(); in XLA_TEST_F()
Dreduce_test.cc113 client_->TransferToServer(input_literal).ConsumeValueOrDie(); in RunR1ToR0Test()
141 client_->TransferToServer(input_literal).ConsumeValueOrDie(); in RunR1ToR0PredTest()
184 client_->TransferToServer(input_literal).ConsumeValueOrDie(); in RunR2ToR1PredTest()
217 client_->TransferToServer(input_literal).ConsumeValueOrDie(); in RunR2ToR0Test()
244 client_->TransferToServer(input_literal).ConsumeValueOrDie(); in RunR2ToR1Test()
301 client_->TransferToServer(input_literal).ConsumeValueOrDie(); in RunVectorizedReduceTestForType()
454 client_->TransferToServer(input_literal).ConsumeValueOrDie(); in XLA_TEST_F()
485 client_->TransferToServer(input_literal).ConsumeValueOrDie(); in XLA_TEST_F()
531 client_->TransferToServer(input_literal).ConsumeValueOrDie(); in XLA_TEST_F()
823 client_->TransferToServer(input_literal).ConsumeValueOrDie(); in XLA_TEST_P()
[all …]
Dscalar_computations_test.cc166 client_->TransferToServer(a_literal).ConsumeValueOrDie(); in XLA_TEST_F()
243 client_->TransferToServer(a_literal).ConsumeValueOrDie(); in XLA_TEST_F()
245 client_->TransferToServer(b_literal).ConsumeValueOrDie(); in XLA_TEST_F()
247 client_->TransferToServer(c_literal).ConsumeValueOrDie(); in XLA_TEST_F()
390 client_->TransferToServer(dividend_literal)); in XLA_TEST_F()
392 client_->TransferToServer(divisor_literal)); in XLA_TEST_F()
432 client_->TransferToServer(dividend_literal)); in XLA_TEST_F()
434 client_->TransferToServer(divisor_literal)); in XLA_TEST_F()
455 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()
371 client_->TransferToServer(MaybeConvertLiteralToBfloat16(argument))); in ComputeAndCompareLiteralWithStatus()
499 TF_ASSIGN_OR_RETURN(auto data, client_->TransferToServer(arg.Clone())); in ComputeValueAndReference()
500 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.cc96 client_->TransferToServer(LiteralUtil::CreateR0<int32>(2)) in XLA_TEST_F()
99 client_->TransferToServer(LiteralUtil::CreateR0<int32>(3)) in XLA_TEST_F()
Dclient_library_test_base.h613 client_->TransferToServer(literal).ConsumeValueOrDie(); in CreateR0Parameter()
627 client_->TransferToServer(literal).ConsumeValueOrDie(); in CreateR1Parameter()
641 client_->TransferToServer(literal).ConsumeValueOrDie(); in CreateR2Parameter()
655 client_->TransferToServer(literal).ConsumeValueOrDie(); in CreateR3Parameter()
669 client_->TransferToServer(literal).ConsumeValueOrDie(); in CreateR4Parameter()
683 client_->TransferToServer(literal).ConsumeValueOrDie(); in CreateParameter()
Dexecution_profile_test.cc33 client_->TransferToServer( in XLA_TEST_F()
Dmatrix_ops_simple_test.cc203 client_->TransferToServer(LiteralUtil::CreateR2FromArray2DWithLayout<T>( in TestImpl()
207 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()
Dxla_compiler_test.cc224 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in TEST_F()
226 client_->TransferToServer(param1_literal).ConsumeValueOrDie(); in TEST_F()
275 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in TEST_F()
277 client_->TransferToServer(param1_literal).ConsumeValueOrDie(); in TEST_F()
612 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in TEST_F()
843 client_->TransferToServer(input).ConsumeValueOrDie(); in TEST_F()
1028 client->TransferToServer(param0_literal).ConsumeValueOrDie(); in RunAndCheckVariablesComputation()
1030 client->TransferToServer(param1_literal).ConsumeValueOrDie(); in RunAndCheckVariablesComputation()
1190 client_->TransferToServer(param1_literal).ConsumeValueOrDie(); in TEST_F()
1310 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in TEST_F()
[all …]
/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/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()
/external/tensorflow/tensorflow/compiler/xla/service/
Dcompile_only_service.h70 Status TransferToServer(const TransferToServerRequest* arg, in TransferToServer() function

12