Home
last modified time | relevance | path

Searched refs:EXPECT_IS_OK (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module_metadata_test.cc56 EXPECT_IS_OK(module_metadata.RecordPassEnd()); in TEST()
68 EXPECT_IS_OK(module_metadata.RecordPassEnd()); in TEST()
74 EXPECT_IS_OK(module_metadata.RecordPassEnd()); in TEST()
86 EXPECT_IS_OK(module_metadata.RecordPassEnd()); in TEST()
94 EXPECT_IS_OK(module_metadata.set_current_pass_name("fake name")); in TEST()
104 EXPECT_IS_OK(module_metadata.set_current_pass_name("fake name")); in TEST()
120 EXPECT_IS_OK(old_module_metadata.set_current_pass_name("outer pass")); in TEST()
123 EXPECT_IS_OK(old_module_metadata.set_current_pass_name("finished pass")); in TEST()
124 EXPECT_IS_OK(old_module_metadata.RecordPassEnd()); in TEST()
127 EXPECT_IS_OK(old_module_metadata.set_current_pass_name("inner pass")); in TEST()
Dtranspose_folding_test.cc56 EXPECT_IS_OK(folded.status()); in FoldTranspose()
247 EXPECT_IS_OK(conv_shape); in TEST_F()
304 EXPECT_IS_OK(conv_shape); in TEST_F()
366 EXPECT_IS_OK(conv_shape); in TEST_F()
434 EXPECT_IS_OK(conv_shape); in TEST_F()
Dhlo_sharding_test.cc66 EXPECT_IS_OK(sharding.Validate(ShapeUtil::MakeShape(U32, {4}), in TEST_F()
82 EXPECT_IS_OK(sharding.Validate(ShapeUtil::MakeShape(U32, {4}), in TEST_F()
132 EXPECT_IS_OK(sharding.Validate(ShapeUtil::MakeShape(F32, {3, 5}), in TEST_F()
183 EXPECT_IS_OK(tuple_sharding.Validate(nested_tuple_shape, /*num_devices=*/5)); in TEST_F()
Dshape_inference_test.cc69 EXPECT_IS_OK(inferred_status.status()); in ExpectInferredReduceShape()
946 EXPECT_IS_OK(inferred_status.status()); in TEST_F()
955 EXPECT_IS_OK(inferred_status_r1f32.status()); in TEST_F()
962 EXPECT_IS_OK(inferred_status_r1f32_one.status()); in TEST_F()
969 EXPECT_IS_OK(inferred_status_r2s32.status()); in TEST_F()
1017 EXPECT_IS_OK(inferred_status.status()); in TEST_F()
1097 EXPECT_IS_OK(inferred_status.status()); in TEST_F()
1121 EXPECT_IS_OK(inferred_status.status()); in TEST_F()
2139 EXPECT_IS_OK(inferred_status0.status()); in TEST_F()
2146 EXPECT_IS_OK(inferred_status1.status()); in TEST_F()
[all …]
Dbfloat16_normalization_test.cc84 EXPECT_IS_OK(result.status()); in Normalize()
88 EXPECT_IS_OK(verifier.Run(module).status()); in Normalize()
Dbfloat16_conversion_folding_test.cc78 EXPECT_IS_OK(result.status()); in FoldConversions()
Dlayout_assignment_test.cc59 EXPECT_IS_OK(layout_assignment.Run(m).status()); in AssignLayouts()
571 EXPECT_IS_OK(layout_assignment.Run(m.get()).status()); in TEST_F()
Dhlo_computation_test.cc254 EXPECT_IS_OK(computation->Accept(&visitor)); in TEST_F()
Dbfloat16_propagation_test.cc70 EXPECT_IS_OK(result.status()); in PropagatePrecision()
Dhlo_instruction_test.cc1052 EXPECT_IS_OK(add->Accept(&visitor)); in TEST_F()
Dbuffer_assignment_test.cc2060 EXPECT_IS_OK(copy_insertion.Run(module).status()); in RunCopyInsertion()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dcall_test.cc40 EXPECT_IS_OK(build_status.status()); in CreateR0F32IdentityComputation()
50 EXPECT_IS_OK(build_status.status()); in CreateR1S0F32AdditionComputation()
60 EXPECT_IS_OK(build_status.status()); in CreateR1S2F32AdditionComputation()
68 EXPECT_IS_OK(build_status.status()); in CreateR0F32TupleComputation()
Dconditional_test.cc33 EXPECT_IS_OK(build_status.status()); in CreateR0ConstantComputation()
41 EXPECT_IS_OK(build_status.status()); in CreateR0IdentityComputation()
50 EXPECT_IS_OK(build_status.status()); in CreateCeilComputation()
67 EXPECT_IS_OK(build_status.status()); in CreateFloorComputation()
89 EXPECT_IS_OK(build_status.status()); in CreateTupleCeilComputation()
111 EXPECT_IS_OK(build_status.status()); in CreateTupleFloorComputation()
131 EXPECT_IS_OK(build_status.status()); in CreateTupleAddComputation()
151 EXPECT_IS_OK(build_status.status()); in CreateTupleSubComputation()
512 EXPECT_IS_OK(true_builder_result.status()); in XLA_TEST_F()
523 EXPECT_IS_OK(false_builder_result.status()); in XLA_TEST_F()
[all …]
Dclient_library_test_base.cc173 EXPECT_IS_OK(ComputeAndCompareLiteralWithStatus(builder, expected, arguments, in ComputeAndCompareLiteral()
181 EXPECT_IS_OK(ComputeAndCompareLiteralWithStatus(builder, expected, arguments, in ComputeAndCompareLiteral()
419 EXPECT_IS_OK(actual_status.status()); in ComputeAndCompareR1U8()
438 EXPECT_IS_OK(actual_status.status()); in ComputeAndCompareTuple()
450 EXPECT_IS_OK(actual_status.status()); in ComputeAndCompareTuple()
461 EXPECT_IS_OK(status_or_data); in ComputeAndCompare()
473 EXPECT_IS_OK(status_or_data); in ComputeAndCompare()
Dllvm_compiler_test.cc146 EXPECT_IS_OK(compiler->Compile(std::move(module_group), in TestMultiModuleCompilation()
/external/tensorflow/tensorflow/compiler/xla/
Dstatus_macros_test.cc56 EXPECT_IS_OK(status); in TEST()
83 EXPECT_IS_OK(rc); in TEST()
99 EXPECT_IS_OK(status); in TEST()
Dlayout_util_test.cc69 EXPECT_IS_OK(LayoutUtil::CopyLayoutBetweenShapes(src, &dst)); in TEST_F()
75 EXPECT_IS_OK(LayoutUtil::CopyLayoutBetweenShapes(src, &dst)); in TEST_F()
82 EXPECT_IS_OK(LayoutUtil::CopyLayoutBetweenShapes(src, &dst)); in TEST_F()
102 EXPECT_IS_OK(LayoutUtil::CopyLayoutBetweenShapes(src, &dst)); in TEST_F()
162 EXPECT_IS_OK(LayoutUtil::CopyLayoutBetweenShapes(src, &dst)); in TEST_F()
173 EXPECT_IS_OK(LayoutUtil::CopyLayoutBetweenShapes(src, &dst)); in TEST_F()
192 EXPECT_IS_OK(LayoutUtil::CopyLayoutBetweenShapes(src, &dst)); in TEST_F()
Dtest_helpers.h56 #define EXPECT_IS_OK(expression) \ macro
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder_test.cc1267 EXPECT_IS_OK(b.SetInstructionFrontendAttribute(op, "attr_c", "c")); in TEST_F()
1279 EXPECT_IS_OK(b.SetInstructionFrontendAttribute(op, "attr_a", "a2")); in TEST_F()
1302 EXPECT_IS_OK(b.SetInstructionFrontendAttribute(op, "attr_d", "d")); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment_test.cc60 EXPECT_IS_OK(layout_assignment.Run(module).status()); in AssignLayouts()