/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | fusion_bitcast_lift_test.cc | 75 auto module = ParseAndReturnVerifiedModule(hlo_text).ValueOrDie(); in TEST_F() 76 EXPECT_TRUE(FusionBitcastLift().Run(module.get()).ValueOrDie()); in TEST_F() 78 EXPECT_TRUE(HloDCE().Run(module.get()).ValueOrDie()); in TEST_F() 94 EXPECT_TRUE(filecheck_result.ValueOrDie()); in TEST_F() 138 auto module = ParseAndReturnVerifiedModule(hlo_text).ValueOrDie(); in TEST_F() 139 EXPECT_TRUE(FusionBitcastLift().Run(module.get()).ValueOrDie()); in TEST_F() 141 EXPECT_TRUE(HloDCE().Run(module.get()).ValueOrDie()); in TEST_F() 162 EXPECT_TRUE(filecheck_result.ValueOrDie()); in TEST_F() 195 auto module = ParseAndReturnVerifiedModule(hlo_text).ValueOrDie(); in TEST_F() 196 EXPECT_TRUE(FusionBitcastLift().Run(module.get()).ValueOrDie()); in TEST_F() [all …]
|
D | multi_output_fusion_test.cc | 79 .ValueOrDie(); in TEST_F() 80 ASSERT_TRUE(GpuMultiOutputFusion().Run(module.get()).ValueOrDie()); in TEST_F() 112 .ValueOrDie(); in TEST_F() 113 ASSERT_FALSE(GpuMultiOutputFusion().Run(module.get()).ValueOrDie()); in TEST_F() 139 .ValueOrDie(); in TEST_F() 140 ASSERT_FALSE(GpuMultiOutputFusion().Run(module.get()).ValueOrDie()); in TEST_F() 167 .ValueOrDie(); in TEST_F() 168 ASSERT_TRUE(GpuMultiOutputFusion().Run(module.get()).ValueOrDie()); in TEST_F() 200 .ValueOrDie(); in TEST_F() 201 ASSERT_TRUE(GpuMultiOutputFusion().Run(module.get()).ValueOrDie()); in TEST_F() [all …]
|
D | fusion_merger_test.cc | 87 .ValueOrDie(); in TEST_F() 88 EXPECT_TRUE(FusionMerger().Run(module.get()).ValueOrDie()); in TEST_F() 149 .ValueOrDie(); in TEST_F() 152 EXPECT_FALSE(FusionMerger().Run(module.get()).ValueOrDie()); in TEST_F() 195 .ValueOrDie(); in TEST_F() 198 EXPECT_TRUE(FusionMerger().Run(module.get()).ValueOrDie()); in TEST_F() 231 .ValueOrDie(); in TEST_F() 232 EXPECT_TRUE(FusionMerger().Run(module.get()).ValueOrDie()); in TEST_F() 279 .ValueOrDie(); in TEST_F() 280 EXPECT_TRUE(FusionMerger().Run(module.get()).ValueOrDie()); in TEST_F() [all …]
|
D | cudnn_pad_for_convolutions_test.cc | 45 .ValueOrDie(); in TEST_F() 46 EXPECT_TRUE(CudnnPadForConvolutions({7, 0}).Run(module.get()).ValueOrDie()); in TEST_F() 70 .ValueOrDie(); in TEST_F() 71 EXPECT_TRUE(CudnnPadForConvolutions({7, 0}).Run(module.get()).ValueOrDie()); in TEST_F() 92 .ValueOrDie(); in TEST_F() 93 EXPECT_TRUE(CudnnPadForConvolutions({7, 0}).Run(module.get()).ValueOrDie()); in TEST_F() 114 .ValueOrDie(); in TEST_F() 115 EXPECT_TRUE(CudnnPadForConvolutions({7, 0}).Run(module.get()).ValueOrDie()); in TEST_F() 136 .ValueOrDie(); in TEST_F() 137 EXPECT_TRUE(CudnnPadForConvolutions({7, 0}).Run(module.get()).ValueOrDie()); in TEST_F() [all …]
|
D | instruction_fusion_test.cc | 50 .ValueOrDie()); in TEST_F() 70 .ValueOrDie()); in TEST_F() 89 .ValueOrDie()); in TEST_F() 108 .ValueOrDie()); in TEST_F() 128 .ValueOrDie()); in TEST_F() 145 .ValueOrDie()); in TEST_F() 166 .ValueOrDie(); in TEST_F() 170 .ValueOrDie()); in TEST_F() 194 .ValueOrDie(); in TEST_F() 198 .ValueOrDie()); in TEST_F() [all …]
|
D | gpu_fusible_test.cc | 44 .ValueOrDie(); in TEST_F() 76 .ValueOrDie(); in TEST_F() 109 .ValueOrDie(); in TEST_F() 129 .ValueOrDie(); in TEST_F() 149 .ValueOrDie(); in TEST_F() 180 .ValueOrDie(); in TEST_F() 205 .ValueOrDie(); in TEST_F() 221 .ValueOrDie(); in TEST_F() 239 .ValueOrDie(); in TEST_F() 259 .ValueOrDie(); in TEST_F() [all …]
|
D | horizontal_loop_fusion_test.cc | 71 .ValueOrDie(); in TEST_F() 73 EXPECT_TRUE(GpuHorizontalLoopFusion().Run(module.get()).ValueOrDie()); in TEST_F() 74 EXPECT_TRUE(HloDCE().Run(module.get()).ValueOrDie()); in TEST_F() 123 .ValueOrDie(); in TEST_F() 125 EXPECT_FALSE(GpuHorizontalLoopFusion().Run(module.get()).ValueOrDie()); in TEST_F() 159 .ValueOrDie(); in TEST_F() 161 EXPECT_FALSE(GpuHorizontalLoopFusion().Run(module.get()).ValueOrDie()); in TEST_F() 187 .ValueOrDie(); in TEST_F() 192 EXPECT_TRUE(fusion.Run(module.get()).ValueOrDie()); in TEST_F() 193 EXPECT_TRUE(GpuHorizontalLoopFusion().Run(module.get()).ValueOrDie()); in TEST_F() [all …]
|
D | cudnn_support_utils_test.cc | 84 .ValueOrDie(); in TEST_F() 112 .ValueOrDie(); in TEST_F() 143 .ValueOrDie(); in TEST_F() 162 .ValueOrDie(); in TEST_F() 180 .ValueOrDie(); in TEST_F() 200 .ValueOrDie(); in TEST_F() 219 .ValueOrDie(); in TEST_F() 237 .ValueOrDie(); in TEST_F() 259 .ValueOrDie(); in TEST_F() 282 .ValueOrDie(); in TEST_F() [all …]
|
D | reduction_splitter_test.cc | 52 .ValueOrDie(); in TEST_F() 53 ASSERT_TRUE(ReductionSplitter().Run(module.get()).ValueOrDie()); in TEST_F() 82 .ValueOrDie(); in TEST_F() 83 ASSERT_TRUE(ReductionSplitter().Run(module.get()).ValueOrDie()); in TEST_F() 113 .ValueOrDie(); in TEST_F() 114 EXPECT_FALSE(ReductionSplitter().Run(module.get()).ValueOrDie()); in TEST_F() 134 .ValueOrDie(); in TEST_F() 135 EXPECT_FALSE(ReductionSplitter().Run(module.get()).ValueOrDie()); in TEST_F()
|
D | matmul_utils_test.cc | 33 Shape shape = ParseShape("f32[1,2,3,4,5,6]").ValueOrDie(); in TEST() 53 Shape shape = ParseShape(params.shape).ValueOrDie(); in TEST_P() 56 IsOkAndHolds(ParseShape(params.expected_shape).ValueOrDie())); in TEST_P() 76 Shape shape = ParseShape("f32[3,4,5,6,7,8]{5,4,3,2,1,0}").ValueOrDie(); in TEST() 84 Shape shape = ParseShape("f32[3,4,5,6]{3,2,0,1}").ValueOrDie(); in TEST() 107 Shape shape = ParseShape(params.shape).ValueOrDie(); in TEST_P() 108 MatrixLayout result = MatrixLayout::For(shape).ValueOrDie(); in TEST_P() 129 Shape shape = ParseShape("f32[3,4,5]{0,2,1}").ValueOrDie(); in TEST()
|
D | cublas_pad_for_gemms_test.cc | 35 return CublasPadForGemms(PrimitiveType::F16, 8).Run(module).ValueOrDie(); in PadForF16Gemms() 50 .ValueOrDie(); in TEST_F() 85 .ValueOrDie(); in TEST_F() 88 CublasPadForGemms(PrimitiveType::S8, 4).Run(module.get()).ValueOrDie()); in TEST_F() 125 .ValueOrDie(); in TEST_F() 187 .ValueOrDie(); in TEST_F() 220 .ValueOrDie(); in TEST_F() 235 .ValueOrDie(); in TEST_F() 250 .ValueOrDie(); in TEST_F() 265 .ValueOrDie(); in TEST_F() [all …]
|
D | alias_passthrough_params_test.cc | 38 .ValueOrDie(); in TEST_F() 39 EXPECT_TRUE(AliasPassthroughParams().Run(module.get()).ValueOrDie()); in TEST_F() 54 .ValueOrDie(); in TEST_F() 55 EXPECT_TRUE(AliasPassthroughParams().Run(module.get()).ValueOrDie()); in TEST_F() 71 .ValueOrDie(); in TEST_F() 80 EXPECT_TRUE(AliasPassthroughParams().Run(module.get()).ValueOrDie()); in TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | while_loop_simplifier_test.cc | 88 return ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in MakeModuleWithSimpleLoop() 126 return ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in MakeModuleWithSimpleLoopTupleElementLoopBound() 131 ASSERT_TRUE(WhileLoopSimplifier().Run(m.get()).ValueOrDie()); in TEST_F() 139 ASSERT_TRUE(WhileLoopSimplifier().Run(m.get()).ValueOrDie()); in TEST_F() 146 ASSERT_TRUE(WhileLoopSimplifier().Run(m.get()).ValueOrDie()); in TEST_F() 154 ASSERT_TRUE(WhileLoopSimplifier().Run(m.get()).ValueOrDie()); in TEST_F() 161 EXPECT_FALSE(WhileLoopSimplifier().Run(m.get()).ValueOrDie()); in TEST_F() 174 ASSERT_TRUE(WhileLoopSimplifier().Run(m.get()).ValueOrDie()); in TEST_F() 195 EXPECT_FALSE(WhileLoopSimplifier().Run(m.get()).ValueOrDie()); in TEST_F() 209 EXPECT_FALSE(WhileLoopSimplifier().Run(m.get()).ValueOrDie()); in TEST_F() [all …]
|
D | instruction_fusion_test.cc | 56 .ValueOrDie(); in TEST_F() 80 .ValueOrDie(); in TEST_F() 100 .ValueOrDie(); in TEST_F() 134 .ValueOrDie()) in TEST_F() 159 .ValueOrDie(); in TEST_F() 164 .ValueOrDie()) in TEST_F() 189 .ValueOrDie(); in TEST_F() 194 .ValueOrDie()) in TEST_F() 224 .ValueOrDie(); in TEST_F() 230 .ValueOrDie()) in TEST_F() [all …]
|
D | conditional_simplifier_test.cc | 105 ASSERT_TRUE(ConditionalSimplifier().Run(m.get()).ValueOrDie()); in TEST_F() 113 ASSERT_TRUE(ConditionalSimplifier().Run(m.get()).ValueOrDie()); in TEST_F() 129 EXPECT_FALSE(ConditionalSimplifier().Run(m.get()).ValueOrDie()); in TEST_F() 145 EXPECT_FALSE(ConditionalSimplifier().Run(m.get()).ValueOrDie()); in TEST_F() 159 EXPECT_FALSE(ConditionalSimplifier().Run(m.get()).ValueOrDie()); in TEST_F() 171 EXPECT_FALSE(ConditionalSimplifier().Run(m.get()).ValueOrDie()); in TEST_F() 210 EXPECT_TRUE(ConditionalSimplifier().Run(module.get()).ValueOrDie()); in TEST_F() 278 EXPECT_TRUE(ConditionalSimplifier().Run(module.get()).ValueOrDie()); in TEST_F() 314 TF_ASSERT_OK(v.Run(status.ValueOrDie().get()).status()); in TEST_F() 316 ConditionalSimplifier().Run(status.ValueOrDie().get()).ValueOrDie()); in TEST_F() [all …]
|
D | hlo_module_dce_test.cc | 108 .ValueOrDie(); in TEST_F() 111 EXPECT_FALSE(dce.Run(module.get()).ValueOrDie()); in TEST_F() 148 .ValueOrDie(); in TEST_F() 151 EXPECT_FALSE(dce.Run(module.get()).ValueOrDie()); in TEST_F() 186 .ValueOrDie(); in TEST_F() 192 EXPECT_TRUE(dce.Run(module.get()).ValueOrDie()); in TEST_F() 230 .ValueOrDie(); in TEST_F() 236 EXPECT_FALSE(dce.Run(module.get()).ValueOrDie()); in TEST_F() 291 .ValueOrDie(); in TEST_F() 299 EXPECT_TRUE(dce.Run(module.get()).ValueOrDie()); in TEST_F() [all …]
|
D | conditional_code_motion_test.cc | 74 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F() 76 ASSERT_TRUE(pass.Run(&*module).ValueOrDie()); in TEST_F() 123 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F() 125 ASSERT_FALSE(pass.Run(&*module).ValueOrDie()); in TEST_F() 159 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F() 161 ASSERT_TRUE(pass.Run(&*module).ValueOrDie()); in TEST_F() 200 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F() 202 ASSERT_TRUE(pass.Run(&*module).ValueOrDie()); in TEST_F() 241 auto module = ParseAndReturnVerifiedModule(hlo_string).ValueOrDie(); in TEST_F() 243 ASSERT_FALSE(pass.Run(&*module).ValueOrDie()); in TEST_F() [all …]
|
D | logistic_expander.cc | 48 .ValueOrDie()) in ExpandLogisticWithTanh() 49 .ValueOrDie(); in ExpandLogisticWithTanh() 53 .ValueOrDie()) in ExpandLogisticWithTanh() 54 .ValueOrDie(); in ExpandLogisticWithTanh() 64 MakeUnaryHlo(HloOpcode::kNegate, operand).ValueOrDie()) in ExpandLogisticWithExp() 65 .ValueOrDie(); in ExpandLogisticWithExp() 67 MakeBinaryHlo(HloOpcode::kAdd, one_constant, exp_instr).ValueOrDie(); in ExpandLogisticWithExp() 69 .ValueOrDie(); in ExpandLogisticWithExp()
|
D | dynamic_dimension_simplifier_test.cc | 65 ASSERT_TRUE(simplifier.Run(m.get()).ValueOrDie()); in TEST_F() 84 ASSERT_FALSE(simplifier.Run(m.get()).ValueOrDie()); in TEST_F() 100 ASSERT_TRUE(simplifier.Run(m.get()).ValueOrDie()); in TEST_F() 118 ASSERT_FALSE(simplifier.Run(m.get()).ValueOrDie()); in TEST_F() 134 ASSERT_FALSE(simplifier.Run(m.get()).ValueOrDie()); in TEST_F() 148 ASSERT_TRUE(simplifier.Run(m.get()).ValueOrDie()); in TEST_F() 164 ASSERT_TRUE(simplifier.Run(m.get()).ValueOrDie()); in TEST_F() 181 ASSERT_FALSE(simplifier.Run(m.get()).ValueOrDie()); in TEST_F() 194 ASSERT_TRUE(simplifier.Run(m.get()).ValueOrDie()); in TEST_F()
|
/external/tensorflow/tensorflow/dtensor/tests/ |
D | tensor_layout_test.cc | 66 .ValueOrDie(); in BatchLayout() 81 EqualsProto(Layout::FromString(layout_str).ValueOrDie().ToProto())); in TEST_F() 89 EqualsProto(Layout::FromString(layout_str).ValueOrDie().ToProto())); in TEST_F() 118 EqualsProto(Mesh::FromString(mesh_str).ValueOrDie().ToProto())); in TEST_F() 143 EXPECT_EQ(layout.device_location(10).ValueOrDie(), offset); in TEST_F() 145 EXPECT_EQ(layout.device_location(18).ValueOrDie(), offset); in TEST_F() 147 EXPECT_EQ(layout.device_location(31).ValueOrDie(), offset); in TEST_F() 156 .ValueOrDie(); in TEST_F() 165 Layout::FromString("sharding_specs:x, mesh:|x=4,y=4|*TPU").ValueOrDie(); in TEST_F() 174 .ValueOrDie(); in TEST_F() [all …]
|
/external/tensorflow/tensorflow/core/platform/ |
D | statusor_test.cc | 83 EXPECT_EQ(null_status.ValueOrDie(), nullptr); in TEST() 101 EXPECT_EQ(0, *thing.ValueOrDie()); in TEST() 102 int* previous = thing.ValueOrDie().get(); in TEST() 106 EXPECT_EQ(0, *thing.ValueOrDie()); in TEST() 107 EXPECT_NE(previous, thing.ValueOrDie().get()); in TEST() 122 ptr = std::move(thing.ValueOrDie()); in TEST() 129 EXPECT_EQ(0, *const_thing.ValueOrDie()); in TEST() 132 const int* const_previous = const_thing.ValueOrDie().get(); in TEST() 135 EXPECT_EQ(0, *const_thing.ValueOrDie()); in TEST() 136 EXPECT_NE(const_previous, const_thing.ValueOrDie().get()); in TEST() [all …]
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | value_inference_test.cc | 85 EXPECT_EQ(value.ValueOrDie(), false); in TEST_F() 94 ComputeDynamismLiteral(computation, &b).ValueOrDie().Get<bool>({0})); in TEST_F() 103 EXPECT_EQ(ComputeDynamismScalar(tuple, &b, {0}).ValueOrDie(), false); in TEST_F() 104 EXPECT_EQ(ComputeDynamismScalar(tuple, &b, {1}).ValueOrDie(), true); in TEST_F() 116 EXPECT_EQ(ComputeDynamismScalar(tuple_2, &b, {0}).ValueOrDie(), false); in TEST_F() 117 EXPECT_EQ(ComputeDynamismScalar(tuple_2, &b, {1}).ValueOrDie(), true); in TEST_F() 126 EXPECT_EQ(ComputeDynamismScalar(result, &b, {}).ValueOrDie(), true); in TEST_F() 138 EXPECT_EQ(ComputeDynamismScalar(result, &b, {}).ValueOrDie(), true); in TEST_F() 163 EXPECT_FALSE(ComputeDynamismLiteral(result, &b).ValueOrDie().Get<bool>({0})); in TEST_F() 164 EXPECT_TRUE(ComputeDynamismLiteral(result, &b).ValueOrDie().Get<bool>({1})); in TEST_F() [all …]
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
D | gpu_kernel_tiling_test.cc | 81 .ValueOrDie(); in TEST_F() 109 .ValueOrDie(); in TEST_F() 153 .ValueOrDie(); in TEST_F() 188 .ValueOrDie(); in TEST_F() 224 .ValueOrDie(); in TEST_F() 252 .ValueOrDie(); in TEST_F() 280 .ValueOrDie(); in TEST_F() 316 .ValueOrDie(); in TEST_F() 348 .ValueOrDie(); in TEST_F() 392 .ValueOrDie(); in TEST_F() [all …]
|
D | gpu_unrolling_test.cc | 52 ParseAndReturnVerifiedModule(kAddModule, config).ValueOrDie(); in TEST_F() 73 ParseAndReturnVerifiedModule(kAddModule, config).ValueOrDie(); in TEST_F() 95 ParseAndReturnVerifiedModule(kAddModule, config).ValueOrDie(); in TEST_F() 128 ParseAndReturnVerifiedModule(kUnfusedAddModule, config).ValueOrDie(); in TEST_F() 159 ParseAndReturnVerifiedModule(kUnfusedAddModule, config).ValueOrDie(); in TEST_F() 193 ParseAndReturnVerifiedModule(kUnfusedAddModule, config).ValueOrDie(); in TEST_F() 227 ParseAndReturnVerifiedModule(kUnfusedAddModule, config).ValueOrDie(); in TEST_F() 252 ParseAndReturnVerifiedModule(kUnfusedAddModule, config).ValueOrDie(); in TEST_F() 291 .ValueOrDie(); in TEST_F()
|
/external/libtextclassifier/native/utils/intents/ |
D | jni-lua.cc | 165 jenv_, status_or_package_name_str.ValueOrDie().get()); in HandleAndroidCallback() 170 PushString(status_or_package_name_std_str.ValueOrDie()); in HandleAndroidCallback() 209 if (!status_or_bundle.ok() || status_or_bundle.ValueOrDie() == nullptr) { in HandleUserRestrictionsCallback() 229 jenv_, status_or_bundle.ValueOrDie().get(), in HandleUserRestrictionsCallback() 230 jni_cache_->bundle_get_boolean, status_or_key.ValueOrDie().get()); in HandleUserRestrictionsCallback() 235 lua_pushboolean(state_, status_or_permission.ValueOrDie()); in HandleUserRestrictionsCallback() 258 status_or_input_str.ValueOrDie().get()); in HandleUrlEncode() 266 JStringToUtf8String(jenv_, status_or_encoded_str.ValueOrDie().get()); in HandleUrlEncode() 271 PushString(status_or_encoded_std_str.ValueOrDie()); in HandleUrlEncode() 291 status_or_url_str.ValueOrDie().get())); in ParseUri() [all …]
|