Home
last modified time | relevance | path

Searched refs:ASSERT_IS_OK (Results 1 – 25 of 26) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_infeed_test.cc48 ASSERT_IS_OK(client_->TransferToInfeed(literal)); in TestInfeedRoundTrip()
159 ASSERT_IS_OK( in TEST_F()
161 ASSERT_IS_OK( in TEST_F()
163 ASSERT_IS_OK( in TEST_F()
165 ASSERT_IS_OK( in TEST_F()
167 ASSERT_IS_OK( in TEST_F()
252 ASSERT_IS_OK(client_->TransferToInfeed( in TEST_F()
255 ASSERT_IS_OK(client_->TransferToInfeed( in TEST_F()
258 ASSERT_IS_OK(client_->TransferToInfeed( in TEST_F()
261 ASSERT_IS_OK(client_->TransferToInfeed( in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis_test.cc148 ASSERT_IS_OK( in TEST_F()
177 ASSERT_IS_OK( in TEST_F()
206 ASSERT_IS_OK( in TEST_F()
229 ASSERT_IS_OK( in TEST_F()
250 ASSERT_IS_OK( in TEST_F()
276 ASSERT_IS_OK( in TEST_F()
305 ASSERT_IS_OK( in TEST_F()
326 ASSERT_IS_OK( in TEST_F()
344 ASSERT_IS_OK( in TEST_F()
363 ASSERT_IS_OK( in TEST_F()
[all …]
Dshape_inference_test.cc107 ASSERT_IS_OK(inferred_status.status()); in TEST_F()
115 ASSERT_IS_OK(inferred_status.status()); in TEST_F()
122 ASSERT_IS_OK(inferred_status.status()); in TEST_F()
130 ASSERT_IS_OK(inferred_status.status()); in TEST_F()
167 ASSERT_IS_OK(inferred_status.status()); in TEST_F()
174 ASSERT_IS_OK(inferred_status.status()); in TEST_F()
181 ASSERT_IS_OK(inferred_status.status()); in TEST_F()
188 ASSERT_IS_OK(inferred_status.status()); in TEST_F()
195 ASSERT_IS_OK(inferred_status.status()); in TEST_F()
202 ASSERT_IS_OK(inferred_status.status()); in TEST_F()
[all …]
Dhlo_instruction_test.cc163 ASSERT_IS_OK(add->Accept(&visitor)); in TEST_F()
202 ASSERT_IS_OK(add->Accept(&visitor)); in TEST_F()
264 ASSERT_IS_OK(addtotal->Accept(&visitor)); in TEST_F()
313 ASSERT_IS_OK(neg2->Accept(&visitor)); in TEST_F()
352 ASSERT_IS_OK(map->Accept(&visitor)); in TEST_F()
396 ASSERT_IS_OK(reduce->Accept(&visitor)); in TEST_F()
427 ASSERT_IS_OK(foo->ReplaceUseWith(add_foofoo, bar)); in TEST_F()
462 ASSERT_IS_OK(foo->ReplaceUseWith(tuple, bar)); in TEST_F()
491 ASSERT_IS_OK(foo->ReplaceUseWith(exp, bar)); in TEST_F()
527 ASSERT_IS_OK(foo->ReplaceAllUsesWith(bar)); in TEST_F()
[all …]
Dcopy_insertion_test.cc80 ASSERT_IS_OK(copy_insertion.Run(module).status()); in InsertCopies()
1378 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
1381 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
1413 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
1416 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
1481 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
1523 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
1565 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
1932 ASSERT_IS_OK(copy_insertion.Run(&module).status()); in BM_SequentialWhiles()
1983 ASSERT_IS_OK(copy_insertion.Run(&module).status()); in BM_ParallelWhiles()
[all …]
Dhlo_reachability_test.cc163 ASSERT_IS_OK(add->RemoveControlDependencyTo(exp)); in TEST_F()
175 ASSERT_IS_OK(constant2->ReplaceUseWith(negate, constant1)); in TEST_F()
Dhlo_computation_test.cc430 ASSERT_IS_OK(add->AddControlDependencyTo(negate)); in TEST_F()
462 ASSERT_IS_OK(computation->RemoveInstructionAndUnusedOperands(dead_add)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tests/
Ddeallocation_test.cc57 ASSERT_IS_OK(client_->Transfer(*global_data).status()); in TEST_F()
59 ASSERT_IS_OK(client_->Unregister(*global_data)); in TEST_F()
72 ASSERT_IS_OK(client_->Unregister(*global_data)); in TEST_F()
85 ASSERT_IS_OK(client_->Unregister(*global_data)); in TEST_F()
99 ASSERT_IS_OK(client_->Unregister(*global_data)); in XLA_TEST_F()
115 ASSERT_IS_OK(client_->Unregister(*global_data)); in XLA_TEST_F()
131 ASSERT_IS_OK(client_->Unregister(*global_data)); in XLA_TEST_F()
Dtransfer_manager_test.cc77 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
93 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
111 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
127 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
143 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
163 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
183 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
197 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
216 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
231 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
[all …]
Dcheck_execution_arity_test.cc51 ASSERT_IS_OK(computation_status.status()); in TEST_F()
58 ASSERT_IS_OK(result_two_args.status()); in TEST_F()
85 ASSERT_IS_OK(computation_status.status()); in XLA_TEST_F()
99 ASSERT_IS_OK(status.status()); in XLA_TEST_F()
Dlocal_client_execute_test.cc763 ASSERT_IS_OK(executable_status); in XLA_TEST_F()
772 ASSERT_IS_OK(local_client_->mutable_backend() in XLA_TEST_F()
861 ASSERT_IS_OK(local_client_->TransferToInfeedLocal( in XLA_TEST_F()
885 ASSERT_IS_OK(local_client_->TransferToInfeedLocal( in XLA_TEST_F()
924 ASSERT_IS_OK( in BM_LocalClientOverhead()
931 ASSERT_IS_OK(executable_status); in BM_LocalClientOverhead()
940 ASSERT_IS_OK(result); in BM_LocalClientOverhead()
946 ASSERT_IS_OK(result); in BM_LocalClientOverhead()
Dvector_ops_simple_test.cc332 ASSERT_IS_OK(computation_status.status()); in XLA_TEST_F()
345 ASSERT_IS_OK(computation_status.status()); in XLA_TEST_F()
362 ASSERT_IS_OK(computation_status.status()); in XLA_TEST_F()
Dquery_inferred_shape_test.cc37 ASSERT_IS_OK(shape_status.status()); in TEST_F()
Dhlo_metadata_test.cc73 ASSERT_IS_OK(executable_status); in TEST_F()
Dllvm_compiler_test.cc41 ASSERT_IS_OK(backend_or_status.status()); in SetUp()
Dmatrix_ops_simple_test.cc84 ASSERT_IS_OK(computation_status.status()); in XLA_TYPED_TEST()
Dmap_test.cc327 ASSERT_IS_OK(embed4_status.status()); in XLA_TEST_F()
457 ASSERT_IS_OK(computation_status.status()); in TEST_F()
Dtuple_test.cc312 ASSERT_IS_OK(computation_status.status()); in XLA_TEST_F()
Ddynamic_ops_test.cc774 ASSERT_IS_OK(transfer_manager->TransferLiteralToDevice( in BM_DynamicSlice()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder_test.cc490 ASSERT_IS_OK(b.SetDynamicBinding(/*dynamic_size_param_num=*/1, in TEST_F()
508 ASSERT_IS_OK(b.SetDynamicBinding(/*dynamic_size_param_num=*/0, in TEST_F()
527 ASSERT_IS_OK(b.SetDynamicBinding(/*dynamic_size_param_num=*/0, in TEST_F()
532 ASSERT_IS_OK(b.SetDynamicBinding(/*dynamic_size_param_num=*/0, in TEST_F()
552 ASSERT_IS_OK(b.SetDynamicBinding(/*dynamic_size_param_num=*/0, in TEST_F()
557 ASSERT_IS_OK(b.SetDynamicBinding(/*dynamic_size_param_num=*/0, in TEST_F()
578 ASSERT_IS_OK(b.SetDynamicBinding(/*dynamic_size_param_num=*/0, in TEST_F()
600 ASSERT_IS_OK(b.SetDynamicBinding(/*dynamic_size_param_num=*/0, in TEST_F()
621 ASSERT_IS_OK(b.SetDynamicBinding(/*dynamic_size_param_num=*/0, in TEST_F()
645 ASSERT_IS_OK(b.SetDynamicBinding(/*dynamic_size_param_num=*/0, in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/
Dtest_helpers.h63 #undef ASSERT_IS_OK
64 #define ASSERT_IS_OK(expression) \ macro
Dtext_literal_writer_test.cc40 ASSERT_IS_OK(TextLiteralWriter::WriteToPath(literal, path)); in TEST()
Dlayout_util_test.cc159 ASSERT_IS_OK(LayoutUtil::CopyLayoutBetweenShapes(src, &dst)); in TEST_F()
166 ASSERT_IS_OK(LayoutUtil::CopyLayoutBetweenShapes(src, &dst)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dinfeed_test.cc42 ASSERT_IS_OK(client_->TransferToInfeed(literal)); in TestInfeedRoundTrip()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_copy_insertion_test.cc59 ASSERT_IS_OK(copy_insertion.Run(module).status()); in InsertCopies()

12