Home
last modified time | relevance | path

Searched refs:ConsumeValueOrDie (Results 1 – 25 of 115) sorted by relevance

12345

/external/tensorflow/tensorflow/compiler/xla/tests/
Dreplay_test.cc44 XlaComputation computation = builder.Build().ConsumeValueOrDie(); in TEST_F()
48 computation.Snapshot().ConsumeValueOrDie(); in TEST_F()
51 XlaComputation replayed = client_->LoadSnapshot(*module).ConsumeValueOrDie(); in TEST_F()
55 client_->GetComputationShape(computation).ConsumeValueOrDie(); in TEST_F()
57 client_->GetComputationShape(replayed).ConsumeValueOrDie(); in TEST_F()
65 .ConsumeValueOrDie(); in TEST_F()
77 XlaComputation computation = builder.Build().ConsumeValueOrDie(); in XLA_TEST_F()
81 computation.Snapshot().ConsumeValueOrDie(); in XLA_TEST_F()
84 XlaComputation replayed = client_->LoadSnapshot(*module).ConsumeValueOrDie(); in XLA_TEST_F()
88 client_->GetComputationShape(computation).ConsumeValueOrDie(); in XLA_TEST_F()
[all …]
Dconvolution_test.cc260 .ConsumeValueOrDie(); in XLA_TEST_F()
263 .ConsumeValueOrDie(); in XLA_TEST_F()
295 .ConsumeValueOrDie(); in RunTest()
298 .ConsumeValueOrDie(); in RunTest()
330 .ConsumeValueOrDie(); in XLA_TEST_F()
333 .ConsumeValueOrDie(); in XLA_TEST_F()
361 .ConsumeValueOrDie(); in XLA_TEST_F()
364 .ConsumeValueOrDie(); in XLA_TEST_F()
397 .ConsumeValueOrDie(); in RunTest()
400 .ConsumeValueOrDie(); in RunTest()
[all …]
Dcompilation_cache_test.cc50 .ConsumeValueOrDie(); in ExecuteComputationR0F32()
65 .ConsumeValueOrDie(); in ExecuteComputationR2F32()
66 Literal result = client_->Transfer(*data_handle).ConsumeValueOrDie(); in ExecuteComputationR2F32()
79 XlaComputation computation = builder.Build().ConsumeValueOrDie(); in XLA_TEST_F()
91 .ConsumeValueOrDie(); in XLA_TEST_F()
94 .ConsumeValueOrDie(); in XLA_TEST_F()
97 .ConsumeValueOrDie(); in XLA_TEST_F()
101 XlaComputation computation = builder.Build().ConsumeValueOrDie(); in XLA_TEST_F()
117 XlaComputation computation_neg = builder_neg.Build().ConsumeValueOrDie(); in XLA_TEST_F()
121 XlaComputation computation_exp = builder_exp.Build().ConsumeValueOrDie(); in XLA_TEST_F()
[all …]
Dmap_test.cc60 return computation_status.ConsumeValueOrDie(); in CreateAdderToOne()
70 return computation_status.ConsumeValueOrDie(); in CreateMax()
82 return computation_status.ConsumeValueOrDie(); in CreateScalarOne()
97 return computation_status.ConsumeValueOrDie(); in CreateMulByTwo()
116 return computation_status.ConsumeValueOrDie(); in CreateAdderToOneTimesItself()
134 return computation_status.ConsumeValueOrDie(); in CreateMapPlusN()
146 return computation_status.ConsumeValueOrDie(); in CreateGt()
165 return computation_status.ConsumeValueOrDie(); in CreateTernaryAdder()
174 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in TEST_F()
188 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in XLA_TEST_F()
[all …]
Dwhile_test.cc60 condition = builder.Build().ConsumeValueOrDie(); in XLA_TEST_F()
70 body = builder.Build().ConsumeValueOrDie(); in XLA_TEST_F()
96 condition = builder.Build().ConsumeValueOrDie(); in XLA_TEST_F()
106 body = builder.Build().ConsumeValueOrDie(); in XLA_TEST_F()
127 condition = builder.Build().ConsumeValueOrDie(); in XLA_TEST_F()
137 body = builder.Build().ConsumeValueOrDie(); in XLA_TEST_F()
159 condition = builder.Build().ConsumeValueOrDie(); in XLA_TEST_F()
168 body = builder.Build().ConsumeValueOrDie(); in XLA_TEST_F()
197 add = builder.Build().ConsumeValueOrDie(); in XLA_TEST_F()
209 condition = builder.Build().ConsumeValueOrDie(); in XLA_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 …]
Ddeconstruct_tuple_test.cc46 XlaComputation computation = builder->Build().ConsumeValueOrDie(); in ExecuteAndCheckTransfer()
49 .ConsumeValueOrDie(); in ExecuteAndCheckTransfer()
66 auto handles = result_status.ConsumeValueOrDie(); in TEST_F()
86 auto handles1 = result_status1.ConsumeValueOrDie(); in TEST_F()
87 auto handles2 = result_status2.ConsumeValueOrDie(); in TEST_F()
117 auto handles = result_status.ConsumeValueOrDie(); in XLA_TEST_F()
139 auto handles = result_status.ConsumeValueOrDie(); in TEST_F()
175 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in XLA_TEST_F()
182 auto handles = result_status.ConsumeValueOrDie(); in XLA_TEST_F()
Dcheck_execution_arity_test.cc46 client_->TransferToServer(param_literal).ConsumeValueOrDie(); in TEST_F()
48 client_->TransferToServer(param_literal).ConsumeValueOrDie(); in TEST_F()
52 auto computation = computation_status.ConsumeValueOrDie(); in TEST_F()
86 auto computation = computation_status.ConsumeValueOrDie(); in XLA_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()
Dround_trip_packed_literal_test.cc43 client_->TransferToServer(original).ConsumeValueOrDie(); in RoundTripToServer()
44 return client_->Transfer(*data).ConsumeValueOrDie(); in RoundTripToServer()
63 reader.Read(ShapeUtil::MakeShape(F32, {2})).ConsumeValueOrDie(); in TEST_F()
91 .ConsumeValueOrDie(); in TEST_F()
124 .ConsumeValueOrDie(); in TEST_F()
Dconditional_test.cc34 return build_status.ConsumeValueOrDie(); in CreateR0ConstantComputation()
42 return build_status.ConsumeValueOrDie(); in CreateR0IdentityComputation()
51 return build_status.ConsumeValueOrDie(); in CreateCeilComputation()
68 return build_status.ConsumeValueOrDie(); in CreateFloorComputation()
90 return build_status.ConsumeValueOrDie(); in CreateTupleCeilComputation()
112 return build_status.ConsumeValueOrDie(); in CreateTupleFloorComputation()
132 return build_status.ConsumeValueOrDie(); in CreateTupleAddComputation()
152 return build_status.ConsumeValueOrDie(); in CreateTupleSubComputation()
357 Call(&builder, inner_builder_result.ConsumeValueOrDie(), in XLA_TEST_F()
529 Conditional(pred, operands, true_builder_result.ConsumeValueOrDie(), operands, in XLA_TEST_F()
[all …]
Ddot_operation_test.cc202 .ConsumeValueOrDie(); in XLA_TYPED_TEST()
206 .ConsumeValueOrDie(); in XLA_TYPED_TEST()
227 .ConsumeValueOrDie(); in TestImpl()
234 .ConsumeValueOrDie(); 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()
475 .ConsumeValueOrDie(); in TestImpl()
482 .ConsumeValueOrDie(); in TestImpl()
507 .ConsumeValueOrDie(); in XLA_TEST_F()
[all …]
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()
Dbatch_normalization_test.cc75 Shape actual_shape = b->GetShape(operand).ConsumeValueOrDie(); in CheckShape()
187 auto input_shape = builder.GetShape(input_activations).ConsumeValueOrDie(); in XLA_TEST_P()
188 auto sum_shape = builder.GetShape(sum).ConsumeValueOrDie(); in XLA_TEST_P()
531 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()
[all …]
Dlocal_client_execute_test.cc128 auto computation = builder.Build().ConsumeValueOrDie(); in XLA_TEST_F()
161 auto computation = builder.Build().ConsumeValueOrDie(); in XLA_TEST_F()
198 auto computation = builder.Build().ConsumeValueOrDie(); in XLA_TEST_F()
226 auto computation = builder.Build().ConsumeValueOrDie(); in XLA_TEST_F()
299 auto computation = builder.Build().ConsumeValueOrDie(); in XLA_TEST_F()
345 auto computation = builder.Build().ConsumeValueOrDie(); in XLA_TEST_F()
377 auto computation = builder.Build().ConsumeValueOrDie(); in XLA_TEST_F()
425 auto computation = builder.Build().ConsumeValueOrDie(); in XLA_TEST_F()
474 auto computation = builder.Build().ConsumeValueOrDie(); in XLA_TEST_F()
523 auto computation = builder.Build().ConsumeValueOrDie(); in XLA_TEST_F()
[all …]
Dcall_test.cc41 return build_status.ConsumeValueOrDie(); in CreateR0F32IdentityComputation()
51 return build_status.ConsumeValueOrDie(); in CreateR1S0F32AdditionComputation()
61 return build_status.ConsumeValueOrDie(); in CreateR1S2F32AdditionComputation()
69 return build_status.ConsumeValueOrDie(); in CreateR0F32TupleComputation()
Dlocal_client_test_base.cc136 .ConsumeValueOrDie(); in LiteralToShapedBuffer()
142 .ConsumeValueOrDie(); in ShapedBufferToLiteral()
162 .ConsumeValueOrDie(); in ExecuteLocallyOrDie()
171 .ConsumeValueOrDie(); in ExecuteLocallyOrDie()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dbuffer_comparator_test.cc30 : backend_(Backend::CreateDefaultBackend().ConsumeValueOrDie()), in BufferComparatorTest()
34 .ConsumeValueOrDie()) {}
47 .ConsumeValueOrDie(); in CompareEqualFloatBuffers()
52 .ConsumeValueOrDie(); in CompareEqualFloatBuffers()
66 .ConsumeValueOrDie() in CompareEqualFloatBuffers()
68 .ConsumeValueOrDie(); in CompareEqualFloatBuffers()
Dcudnn_fused_conv_rewriter_test.cc35 .ConsumeValueOrDie(), in GetOptimizedHlo()
38 .ConsumeValueOrDie() in GetOptimizedHlo()
298 .ConsumeValueOrDie(), in TEST_F()
301 .ConsumeValueOrDie() in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_infeed_test.cc130 condition = builder.Build().ConsumeValueOrDie(); in TEST_F()
142 body = builder.Build().ConsumeValueOrDie(); in TEST_F()
149 auto computation = builder.Build().ConsumeValueOrDie(); in TEST_F()
171 auto result_literal = client_->Transfer(*result).ConsumeValueOrDie(); in TEST_F()
214 condition = builder.Build().ConsumeValueOrDie(); in TEST_F()
233 return builder.Build().ConsumeValueOrDie(); in TEST_F()
249 auto computation = builder.Build().ConsumeValueOrDie(); in TEST_F()
289 auto result_literal = client_->Transfer(*result).ConsumeValueOrDie(); in TEST_F()
/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()
57 xla::XlaComputation computation = computation_status.ConsumeValueOrDie(); in main()
66 xla::Literal actual = result.ConsumeValueOrDie(); in main()
/external/tensorflow/tensorflow/compiler/xla/tools/
Ddumped_computation_to_text.cc50 XlaComputation computation = computation_status.ConsumeValueOrDie(); in RealMain()
54 client->GetComputationShape(computation).ConsumeValueOrDie(); in RealMain()
76 .ConsumeValueOrDie(); in RealMain()
79 .ConsumeValueOrDie(); in RealMain()
/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_liveness_test.cc126 .ConsumeValueOrDie(); in TEST_F()
175 .ConsumeValueOrDie(); in TEST_F()
222 .ConsumeValueOrDie(); in TEST_F()
256 .ConsumeValueOrDie(); in TEST_F()
300 .ConsumeValueOrDie(); in TEST_F()
349 .ConsumeValueOrDie(); in TEST_F()
382 .ConsumeValueOrDie(); in TEST_F()
418 .ConsumeValueOrDie(); in TEST_F()
459 .ConsumeValueOrDie(); in TEST_F()
524 .ConsumeValueOrDie(); in TEST_F()
[all …]
Dbuffer_assignment_test.cc97 .ConsumeValueOrDie(); in RunBufferAssignment()
108 .ConsumeValueOrDie(); in RunBufferAssignmentNoBuffersForConstants()
126 .ConsumeValueOrDie(); in RunBufferAssignmentNoBuffersReuseForAdd()
137 .ConsumeValueOrDie(); in RunColoredBufferAssignment()
151 .ConsumeValueOrDie(); in RunBufferAssignmentWithInstructionSequence()
246 *buffers.GetUniqueTopLevelSlice(hlo).ConsumeValueOrDie().allocation(); in GetAssignedInputAllocation()
264 return *buffers.GetUniqueSlice(hlo, index).ConsumeValueOrDie().allocation(); in GetAllocation()
269 .ConsumeValueOrDie() in GetTopLevelAllocation()
318 assignment.GetUniqueTopLevelSlice(instruction).ConsumeValueOrDie()); in BuffersDistinct()
325 .ConsumeValueOrDie())) { in BuffersDistinct()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler_test.cc217 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in TEST_F()
219 client_->TransferToServer(param1_literal).ConsumeValueOrDie(); in TEST_F()
224 .ConsumeValueOrDie(); in TEST_F()
225 xla::Literal actual_literal = client_->Transfer(*actual).ConsumeValueOrDie(); in TEST_F()
268 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in TEST_F()
270 client_->TransferToServer(param1_literal).ConsumeValueOrDie(); in TEST_F()
275 .ConsumeValueOrDie(); in TEST_F()
276 xla::Literal actual_literal = client_->Transfer(*actual).ConsumeValueOrDie(); in TEST_F()
504 client_->TransferToServer(param0_literal).ConsumeValueOrDie(); in TEST_F()
508 .ConsumeValueOrDie(); in TEST_F()
[all …]
Dxla_context.cc90 return b.Build().ConsumeValueOrDie(); in GetOrCreateMax()
106 return b.Build().ConsumeValueOrDie(); in GetOrCreateMin()
122 return b.Build().ConsumeValueOrDie(); in GetOrCreateAdd()
138 return b.Build().ConsumeValueOrDie(); in GetOrCreateMul()

12345