Home
last modified time | relevance | path

Searched refs:TransferToServer (Results 1 – 25 of 48) 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.cc259 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(input)) in XLA_TEST_F()
262 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(filter)) in XLA_TEST_F()
294 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(input)) in RunTest()
297 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(filter)) in RunTest()
329 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(input)) in XLA_TEST_F()
332 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(filter)) in XLA_TEST_F()
360 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(input)) in XLA_TEST_F()
363 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(filter)) in XLA_TEST_F()
396 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(input)) in RunTest()
399 client_->TransferToServer(LiteralUtil::CreateR3FromArray3D(filter)) in RunTest()
[all …]
Ddot_operation_test.cc200 ->TransferToServer(LiteralUtil::CreateR2FromArray2D<T>( in XLA_TYPED_TEST()
204 ->TransferToServer(LiteralUtil::CreateR2FromArray2D<T>( in XLA_TYPED_TEST()
223 ->TransferToServer(LiteralUtil::CreateFromArrayWithLayout<T>( in TestImpl()
230 ->TransferToServer(LiteralUtil::CreateFromArrayWithLayout<T>( in TestImpl()
294 client_->TransferToServer(dot_lhs_lit).ConsumeValueOrDie(); in TestImpl()
303 client_->TransferToServer(dot_rhs_lit).ConsumeValueOrDie(); in TestImpl()
314 addend_handle = client_->TransferToServer(addend_lit).ConsumeValueOrDie(); in TestImpl()
471 ->TransferToServer(LiteralUtil::CreateFromArrayWithLayout<T>( in TestImpl()
478 ->TransferToServer(LiteralUtil::CreateFromArrayWithLayout<T>( in TestImpl()
505 ->TransferToServer(LiteralUtil::CreateR2WithLayout<complex64>( 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()
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()
Dreduce_precision_test.cc234 client_->TransferToServer(a_literal).ConsumeValueOrDie(); in XLA_TEST_P()
257 client_->TransferToServer(a_literal).ConsumeValueOrDie(); in XLA_TEST_F()
286 client_->TransferToServer(a_literal).ConsumeValueOrDie(); in XLA_TEST_F()
312 client_->TransferToServer(a_literal).ConsumeValueOrDie(); in XLA_TEST_F()
336 client_->TransferToServer(a_literal).ConsumeValueOrDie(); in XLA_TEST_F()
361 client_->TransferToServer(a_literal).ConsumeValueOrDie(); in XLA_TEST_F()
Darray_elementwise_ops_test.cc231 client_->TransferToServer(lhs_literal).ConsumeValueOrDie(); in XLA_TEST_F()
245 client_->TransferToServer(rhs_literal).ConsumeValueOrDie(); in XLA_TEST_F()
271 client_->TransferToServer(lhs_literal).ConsumeValueOrDie(); in XLA_TEST_F()
284 client_->TransferToServer(rhs_literal).ConsumeValueOrDie(); in XLA_TEST_F()
324 client_->TransferToServer(a_literal).ConsumeValueOrDie(); in TEST_P()
330 client_->TransferToServer(b_literal).ConsumeValueOrDie(); in TEST_P()
383 auto a_data = client_->TransferToServer(a_literal).ConsumeValueOrDie(); in XLA_TEST_F()
384 auto b_data = client_->TransferToServer(b_literal).ConsumeValueOrDie(); in XLA_TEST_F()
1482 client_->TransferToServer(param_literal).ConsumeValueOrDie(); in XLA_TEST_F()
1510 client_->TransferToServer(literal0).ConsumeValueOrDie(); in XLA_TEST_F()
[all …]
Dbatch_normalization_test.cc531 client_->TransferToServer(input_literal).ConsumeValueOrDie(); in XLA_TEST_P()
533 client_->TransferToServer(scale_literal).ConsumeValueOrDie(); in XLA_TEST_P()
535 client_->TransferToServer(offset_literal).ConsumeValueOrDie(); in XLA_TEST_P()
632 client_->TransferToServer(input_literal).ConsumeValueOrDie(); in XLA_TEST_P()
634 client_->TransferToServer(scale_literal).ConsumeValueOrDie(); in XLA_TEST_P()
636 client_->TransferToServer(offset_literal).ConsumeValueOrDie(); in XLA_TEST_P()
638 client_->TransferToServer(mean_literal).ConsumeValueOrDie(); in XLA_TEST_P()
640 client_->TransferToServer(var_literal).ConsumeValueOrDie(); in XLA_TEST_P()
817 client_->TransferToServer(input_literal).ConsumeValueOrDie(); in XLA_TEST_P()
819 client_->TransferToServer(scale_literal).ConsumeValueOrDie(); in XLA_TEST_P()
[all …]
Dreduce_test.cc109 client_->TransferToServer(input_literal).ConsumeValueOrDie(); in RunR1ToR0Test()
140 client_->TransferToServer(input_literal).ConsumeValueOrDie(); in RunR1ToR0PredTest()
183 client_->TransferToServer(input_literal).ConsumeValueOrDie(); in RunR2ToR1PredTest()
216 client_->TransferToServer(input_literal).ConsumeValueOrDie(); in RunR2ToR0Test()
243 client_->TransferToServer(input_literal).ConsumeValueOrDie(); in RunR2ToR1Test()
300 client_->TransferToServer(input_literal).ConsumeValueOrDie(); in RunVectorizedReduceTestForType()
453 client_->TransferToServer(input_literal).ConsumeValueOrDie(); in XLA_TEST_F()
484 client_->TransferToServer(input_literal).ConsumeValueOrDie(); in XLA_TEST_F()
530 client_->TransferToServer(input_literal).ConsumeValueOrDie(); in XLA_TEST_F()
822 client_->TransferToServer(input_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()
Dconcat_test.cc517 auto a_data = client_->TransferToServer(a_literal).ConsumeValueOrDie(); in XLA_TEST_F()
565 auto x_data = client_->TransferToServer(x_literal).ConsumeValueOrDie(); in XLA_TEST_F()
566 auto y_data = client_->TransferToServer(y_literal).ConsumeValueOrDie(); in XLA_TEST_F()
588 auto x_data = client_->TransferToServer(x_literal).ConsumeValueOrDie(); in XLA_TEST_F()
589 auto y_data = client_->TransferToServer(y_literal).ConsumeValueOrDie(); in XLA_TEST_F()
590 auto z_data = client_->TransferToServer(z_literal).ConsumeValueOrDie(); in XLA_TEST_F()
616 auto x_data = client_->TransferToServer(x_literal).ConsumeValueOrDie(); in XLA_TEST_F()
617 auto y_data = client_->TransferToServer(y_literal).ConsumeValueOrDie(); in XLA_TEST_F()
618 auto z_data = client_->TransferToServer(z_literal).ConsumeValueOrDie(); in XLA_TEST_F()
Dscalar_computations_test.cc166 client_->TransferToServer(a_literal).ConsumeValueOrDie(); in XLA_TEST_F()
233 client_->TransferToServer(a_literal).ConsumeValueOrDie(); in XLA_TEST_F()
235 client_->TransferToServer(b_literal).ConsumeValueOrDie(); in XLA_TEST_F()
237 client_->TransferToServer(c_literal).ConsumeValueOrDie(); in XLA_TEST_F()
380 client_->TransferToServer(dividend_literal)); in XLA_TEST_F()
382 client_->TransferToServer(divisor_literal)); in XLA_TEST_F()
422 client_->TransferToServer(dividend_literal)); in XLA_TEST_F()
424 client_->TransferToServer(divisor_literal)); in XLA_TEST_F()
445 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()
Dreplay_test.cc96 client_->TransferToServer(LiteralUtil::CreateR0<int32>(2)) in XLA_TEST_F()
99 client_->TransferToServer(LiteralUtil::CreateR0<int32>(3)) in XLA_TEST_F()
Dmatrix_ops_simple_test.cc203 client_->TransferToServer(LiteralUtil::CreateR2FromArray2DWithLayout<T>( in TestImpl()
207 client_->TransferToServer(LiteralUtil::CreateR2FromArray2DWithLayout<T>( in TestImpl()
Dexecution_profile_test.cc33 client_->TransferToServer( in XLA_TEST_F()
Dslice_test.cc236 client_->TransferToServer(literal)); in Run()
402 client_->TransferToServer(literal)); in XLA_TEST_P()
492 client_->TransferToServer(literal)); in Run()
Dclient_library_test_base.h557 client_->TransferToServer(literal).ConsumeValueOrDie(); in CreateR0Parameter()
571 client_->TransferToServer(literal).ConsumeValueOrDie(); in CreateR1Parameter()
585 client_->TransferToServer(literal).ConsumeValueOrDie(); in CreateR2Parameter()
599 client_->TransferToServer(literal).ConsumeValueOrDie(); in CreateR3Parameter()
/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.cc80 auto x_global_or = client->TransferToServer(x_literal); in TEST()
81 auto y_global_or = client->TransferToServer(y_literal); in TEST()
Dxla_compiler_test.cc217 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in TEST_F()
219 client_->TransferToServer(param1_literal).ConsumeValueOrDie(); in TEST_F()
268 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in TEST_F()
270 client_->TransferToServer(param1_literal).ConsumeValueOrDie(); in TEST_F()
504 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in TEST_F()
536 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in TEST_F()
775 client_->TransferToServer(input).ConsumeValueOrDie(); in TEST_F()
963 client->TransferToServer(param0_literal).ConsumeValueOrDie(); in RunAndCheckVariablesComputation()
965 client->TransferToServer(param1_literal).ConsumeValueOrDie(); in RunAndCheckVariablesComputation()
1122 client_->TransferToServer(param1_literal).ConsumeValueOrDie(); in TEST_F()
[all …]
/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