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/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()
115 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
133 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
151 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
169 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
185 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
205 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
225 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
239 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()
Dquery_inferred_shape_test.cc37 ASSERT_IS_OK(shape_status.status()); in TEST_F()
Dvector_ops_simple_test.cc344 ASSERT_IS_OK(computation_status.status()); in XLA_TEST_F()
357 ASSERT_IS_OK(computation_status.status()); in XLA_TEST_F()
374 ASSERT_IS_OK(computation_status.status()); in XLA_TEST_F()
Dlocal_client_execute_test.cc775 ASSERT_IS_OK(local_client_->mutable_backend() in XLA_TEST_F()
912 ASSERT_IS_OK(local_client_->TransferToInfeedLocal( in XLA_TEST_F()
936 ASSERT_IS_OK(local_client_->TransferToInfeedLocal( in XLA_TEST_F()
941 ASSERT_IS_OK(local_client_->TransferFromOutfeedLocal( in XLA_TEST_F()
973 ASSERT_IS_OK( in BM_LocalClientOverhead()
988 ASSERT_IS_OK(result); in BM_LocalClientOverhead()
993 ASSERT_IS_OK(result); in BM_LocalClientOverhead()
Dllvm_compiler_test.cc83 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.cc791 ASSERT_IS_OK(transfer_manager->TransferLiteralToDevice( in BM_DynamicSlice()
Ddot_operation_test.cc1861 ASSERT_IS_OK(executable->Run({&buffer0}, options)); in DOT_ReorderContracting()
1866 ASSERT_IS_OK(executable->Run({&buffer0}, options)); in DOT_ReorderContracting()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis_test.cc147 ASSERT_IS_OK( in TEST_F()
181 ASSERT_IS_OK( in TEST_F()
217 ASSERT_IS_OK( in TEST_F()
247 ASSERT_IS_OK( in TEST_F()
274 ASSERT_IS_OK( in TEST_F()
304 ASSERT_IS_OK( in TEST_F()
338 ASSERT_IS_OK( in TEST_F()
361 ASSERT_IS_OK( in TEST_F()
384 ASSERT_IS_OK( in TEST_F()
413 ASSERT_IS_OK( in TEST_F()
[all …]
Dloop_schedule_linearizer_test.cc77 ASSERT_IS_OK(loop_schedule_linearizer.Run(module).status()); in InsertCopies()
80 ASSERT_IS_OK(copy_insertion.Run(module).status()); in InsertCopies()
Dshape_inference_test.cc109 ASSERT_IS_OK(inferred_status.status()); in TEST_F()
135 ASSERT_IS_OK(inferred_status.status()); in TEST_F()
173 ASSERT_IS_OK(inferred_status.status()); in TEST_F()
180 ASSERT_IS_OK(inferred_status.status()); in TEST_F()
309 ASSERT_IS_OK(result.status()); in TEST_F()
334 ASSERT_IS_OK(inferred_status.status()); in TEST_F()
343 ASSERT_IS_OK(inferred_status_ok.status()); in TEST_F()
441 ASSERT_IS_OK(inferred_status.status()); in TEST_F()
487 ASSERT_IS_OK(inferred_status.status()); in TEST_F()
533 ASSERT_IS_OK(inferred_status.status()); in TEST_F()
[all …]
Dcopy_insertion_test.cc80 ASSERT_IS_OK(copy_insertion.Run(module).status()); in InsertCopies()
1588 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
1590 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
1621 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
1623 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
1687 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
1728 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
1769 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
2139 ASSERT_IS_OK(copy_insertion.Run(&module).status()); in BM_SequentialWhiles()
2193 ASSERT_IS_OK(copy_insertion.Run(&module).status()); in BM_ParallelWhiles()
[all …]
Dhlo_instruction_test.cc165 ASSERT_IS_OK(add->Accept(&visitor)); in TEST_F()
204 ASSERT_IS_OK(add->Accept(&visitor)); in TEST_F()
266 ASSERT_IS_OK(addtotal->Accept(&visitor)); in TEST_F()
315 ASSERT_IS_OK(neg2->Accept(&visitor)); in TEST_F()
354 ASSERT_IS_OK(map->Accept(&visitor)); in TEST_F()
398 ASSERT_IS_OK(reduce->Accept(&visitor)); in TEST_F()
429 ASSERT_IS_OK(foo->ReplaceUseWith(add_foofoo, bar)); in TEST_F()
464 ASSERT_IS_OK(foo->ReplaceUseWith(tuple, bar)); in TEST_F()
493 ASSERT_IS_OK(foo->ReplaceUseWith(exp, bar)); in TEST_F()
529 ASSERT_IS_OK(foo->ReplaceAllUsesWith(bar)); in TEST_F()
[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.cc429 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/client/
Dxla_builder_test.cc562 ASSERT_IS_OK(b.SetDynamicBinding(/*dynamic_size_param_num=*/1, in TEST_F()
606 ASSERT_IS_OK(b.SetDynamicBinding(/*dynamic_size_param_num=*/0, in TEST_F()
625 ASSERT_IS_OK(b.SetDynamicBinding(/*dynamic_size_param_num=*/0, in TEST_F()
630 ASSERT_IS_OK(b.SetDynamicBinding(/*dynamic_size_param_num=*/0, in TEST_F()
650 ASSERT_IS_OK(b.SetDynamicBinding(/*dynamic_size_param_num=*/0, in TEST_F()
655 ASSERT_IS_OK(b.SetDynamicBinding(/*dynamic_size_param_num=*/0, in TEST_F()
676 ASSERT_IS_OK(b.SetDynamicBinding(/*dynamic_size_param_num=*/0, in TEST_F()
698 ASSERT_IS_OK(b.SetDynamicBinding(/*dynamic_size_param_num=*/0, in TEST_F()
719 ASSERT_IS_OK(b.SetDynamicBinding(/*dynamic_size_param_num=*/0, in TEST_F()
743 ASSERT_IS_OK(b.SetDynamicBinding(/*dynamic_size_param_num=*/0, in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/
Dtest_helpers.h60 #undef ASSERT_IS_OK
61 #define ASSERT_IS_OK(expression) \ macro
Dtext_literal_writer_test.cc41 ASSERT_IS_OK(TextLiteralWriter::WriteToPath(literal, path)); in TEST()
Dlayout_util_test.cc109 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()

12