/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_module_metadata_test.cc | 56 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()
|
D | transpose_folding_test.cc | 56 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()
|
D | hlo_sharding_test.cc | 66 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()
|
D | shape_inference_test.cc | 69 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 …]
|
D | bfloat16_normalization_test.cc | 84 EXPECT_IS_OK(result.status()); in Normalize() 88 EXPECT_IS_OK(verifier.Run(module).status()); in Normalize()
|
D | bfloat16_conversion_folding_test.cc | 78 EXPECT_IS_OK(result.status()); in FoldConversions()
|
D | layout_assignment_test.cc | 59 EXPECT_IS_OK(layout_assignment.Run(m).status()); in AssignLayouts() 571 EXPECT_IS_OK(layout_assignment.Run(m.get()).status()); in TEST_F()
|
D | hlo_computation_test.cc | 254 EXPECT_IS_OK(computation->Accept(&visitor)); in TEST_F()
|
D | bfloat16_propagation_test.cc | 70 EXPECT_IS_OK(result.status()); in PropagatePrecision()
|
D | hlo_instruction_test.cc | 1052 EXPECT_IS_OK(add->Accept(&visitor)); in TEST_F()
|
D | buffer_assignment_test.cc | 2060 EXPECT_IS_OK(copy_insertion.Run(module).status()); in RunCopyInsertion()
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | call_test.cc | 40 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()
|
D | conditional_test.cc | 33 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 …]
|
D | client_library_test_base.cc | 173 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()
|
D | llvm_compiler_test.cc | 146 EXPECT_IS_OK(compiler->Compile(std::move(module_group), in TestMultiModuleCompilation()
|
/external/tensorflow/tensorflow/compiler/xla/ |
D | status_macros_test.cc | 56 EXPECT_IS_OK(status); in TEST() 83 EXPECT_IS_OK(rc); in TEST() 99 EXPECT_IS_OK(status); in TEST()
|
D | layout_util_test.cc | 69 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()
|
D | test_helpers.h | 56 #define EXPECT_IS_OK(expression) \ macro
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | xla_builder_test.cc | 1267 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/ |
D | cpu_layout_assignment_test.cc | 60 EXPECT_IS_OK(layout_assignment.Run(module).status()); in AssignLayouts()
|