/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | broadcast_test.cc | 45 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() 61 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() 84 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() 105 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() 124 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() 141 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() 161 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() 186 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() 217 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() 233 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() [all …]
|
D | cpu_gpu_fusion_test.cc | 91 auto hlo_module = CreateNewVerifiedModule(); in TestElementwise2D() 201 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() 249 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() 284 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() 308 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() 332 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() 347 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() 362 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() 377 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() 392 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() [all …]
|
D | custom_call_test.cc | 82 auto module = CreateNewVerifiedModule(); in XLA_TEST_F() 97 auto module = CreateNewVerifiedModule(); in XLA_TEST_F() 118 auto module = CreateNewVerifiedModule(); in XLA_TEST_F() 142 auto module = CreateNewVerifiedModule(); in XLA_TEST_F() 167 auto module = CreateNewVerifiedModule(); in XLA_TEST_F()
|
D | copy_test.cc | 48 auto module = CreateNewVerifiedModule(); in TestCopyOp() 101 auto module = CreateNewVerifiedModule(); in XLA_TEST_F() 122 auto module = CreateNewVerifiedModule(); in XLA_TEST_F() 148 auto module = CreateNewVerifiedModule(); in XLA_TEST_F() 180 auto module = CreateNewVerifiedModule(); in TestCopyConstantLayout021() 214 auto module = CreateNewVerifiedModule(); in TestCopyConstantLayoutR4()
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | cpu_instruction_fusion_test.cc | 62 auto module = CreateNewVerifiedModule(); in TEST_F() 81 auto module = CreateNewVerifiedModule(); in TEST_F() 102 auto module = CreateNewVerifiedModule(); in TEST_F() 124 auto module = CreateNewVerifiedModule(); in TEST_F() 143 auto module = CreateNewVerifiedModule(); in TEST_F() 162 auto module = CreateNewVerifiedModule(); in TEST_F() 326 auto module = CreateNewVerifiedModule(); in TEST_F() 357 auto module = CreateNewVerifiedModule(); in TEST_F() 378 auto module = CreateNewVerifiedModule(); in TEST_F() 400 auto module = CreateNewVerifiedModule(); in TEST_F() [all …]
|
D | cpu_layout_assignment_test.cc | 76 auto module = CreateNewVerifiedModule(); in TEST_F() 117 auto module = CreateNewVerifiedModule(); in TEST_F() 161 auto module = CreateNewVerifiedModule(); in TEST_F() 195 auto module = CreateNewVerifiedModule(); in TEST_F() 235 auto module = CreateNewVerifiedModule(); in TEST_F() 368 std::unique_ptr<HloModule> module = CreateNewVerifiedModule(); in TEST_F() 380 std::unique_ptr<HloModule> module = CreateNewVerifiedModule(); in TEST_F() 392 std::unique_ptr<HloModule> module = CreateNewVerifiedModule(); in TEST_F() 404 std::unique_ptr<HloModule> module = CreateNewVerifiedModule(); in TEST_F() 416 std::unique_ptr<HloModule> module = CreateNewVerifiedModule(); in TEST_F() [all …]
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_computation_test.cc | 74 auto module = CreateNewVerifiedModule(); in TEST_F() 82 auto module = CreateNewVerifiedModule(); in TEST_F() 94 auto module = CreateNewVerifiedModule(); in TEST_F() 128 auto module = CreateNewVerifiedModule(); in TEST_F() 143 auto module = CreateNewVerifiedModule(); in TEST_F() 160 auto module = CreateNewVerifiedModule(); in TEST_F() 179 auto module = CreateNewVerifiedModule(); in TEST_F() 201 auto module = CreateNewVerifiedModule(); in TEST_F() 226 auto module = CreateNewVerifiedModule(); in TEST_F() 266 auto module = CreateNewVerifiedModule(); in TEST_F() [all …]
|
D | tuple_simplifier_test.cc | 68 auto module = CreateNewVerifiedModule(); in TEST_F() 81 auto module = CreateNewVerifiedModule(); in TEST_F() 101 auto module = CreateNewVerifiedModule(); in TEST_F() 128 auto module = CreateNewVerifiedModule(); in TEST_F() 160 auto module = CreateNewVerifiedModule(); in TEST_F() 185 auto module = CreateNewVerifiedModule(); in TEST_F() 210 auto module = CreateNewVerifiedModule(); in TEST_F() 222 auto module = CreateNewVerifiedModule(); in TEST_F()
|
D | hlo_module_test.cc | 67 auto module = CreateNewVerifiedModule(); in TEST_F() 76 auto module = CreateNewVerifiedModule(); in TEST_F() 92 auto module = CreateNewVerifiedModule(); in TEST_F() 115 auto module = CreateNewVerifiedModule(); in TEST_F() 158 auto module = CreateNewVerifiedModule(); in TEST_F() 178 auto module = CreateNewVerifiedModule(); in TEST_F() 198 auto module_a = CreateNewVerifiedModule(); in TEST_F() 199 auto module_b = CreateNewVerifiedModule(); in TEST_F()
|
D | hlo_dce_test.cc | 62 auto module = CreateNewVerifiedModule(); in TEST_F() 84 auto module = CreateNewVerifiedModule(); in TEST_F() 105 auto module = CreateNewVerifiedModule(); in TEST_F() 122 auto module = CreateNewVerifiedModule(); in TEST_F() 149 auto module = CreateNewVerifiedModule(); in TEST_F() 189 auto module = CreateNewVerifiedModule(); in TEST_F() 214 auto module = CreateNewVerifiedModule(); in TEST_F() 254 auto module = CreateNewVerifiedModule(); in TEST_F() 312 auto module = CreateNewVerifiedModule(); in TEST_F() 363 auto module = CreateNewVerifiedModule(); in TEST_F() [all …]
|
D | algebraic_simplifier_test.cc | 59 auto m = CreateNewVerifiedModule(); in TEST_F() 282 auto m = CreateNewVerifiedModule(); in TEST_F() 377 auto module = CreateNewVerifiedModule(); in TEST_F() 399 auto module = CreateNewVerifiedModule(); in TEST_F() 419 auto module = CreateNewVerifiedModule(); in TEST_F() 430 auto m = CreateNewVerifiedModule(); in TEST_F() 468 auto m = CreateNewVerifiedModule(); in TEST_F() 489 auto m = CreateNewVerifiedModule(); in TEST_F() 537 auto m = CreateNewVerifiedModule(); in TEST_F() 559 auto m = CreateNewVerifiedModule(); in TEST_F() [all …]
|
D | reshape_mover_test.cc | 40 auto m = CreateNewVerifiedModule(); in TEST_F() 78 auto m = CreateNewVerifiedModule(); 109 auto m = CreateNewVerifiedModule(); in TEST_F() 136 auto m = CreateNewVerifiedModule(); in TEST_F() 184 auto m = CreateNewVerifiedModule(); 229 auto m = CreateNewVerifiedModule(); 266 auto m = CreateNewVerifiedModule(); 319 auto m = CreateNewVerifiedModule(); 359 auto m = CreateNewVerifiedModule(); 388 auto m = CreateNewVerifiedModule(); in TEST_F() [all …]
|
D | hlo_instruction_test.cc | 157 auto module = CreateNewVerifiedModule(); in TEST_F() 194 auto module = CreateNewVerifiedModule(); in TEST_F() 227 auto module = CreateNewVerifiedModule(); in TEST_F() 262 auto module = CreateNewVerifiedModule(); in TEST_F() 311 auto module = CreateNewVerifiedModule(); in TEST_F() 333 auto module = CreateNewVerifiedModule(); in TEST_F() 381 auto module = CreateNewVerifiedModule(); in TEST_F() 422 auto module = CreateNewVerifiedModule(); in TEST_F() 457 auto module = CreateNewVerifiedModule(); in TEST_F() 485 auto module = CreateNewVerifiedModule(); in TEST_F() [all …]
|
D | hlo_cse_test.cc | 62 auto module = CreateNewVerifiedModule(); in TEST_F() 92 auto module = CreateNewVerifiedModule(); in TEST_F() 124 auto module = CreateNewVerifiedModule(); in TEST_F() 174 auto module = CreateNewVerifiedModule(); in TEST_F() 204 auto module = CreateNewVerifiedModule(); in TEST_F() 236 auto module = CreateNewVerifiedModule(); in TEST_F() 442 auto module = CreateNewVerifiedModule(); in TEST_F() 473 auto module = CreateNewVerifiedModule(); in TEST_F() 491 auto module = CreateNewVerifiedModule(); in TEST_F() 557 auto module = CreateNewVerifiedModule(); in TEST_F() [all …]
|
D | bfloat16_normalization_test.cc | 112 auto module = CreateNewVerifiedModule(); in TEST_F() 140 auto module = CreateNewVerifiedModule(); in TEST_F() 170 auto module = CreateNewVerifiedModule(); in TEST_F() 197 auto module = CreateNewVerifiedModule(); in TEST_F() 239 auto module = CreateNewVerifiedModule(); in TEST_F() 277 auto module = CreateNewVerifiedModule(); in TEST_F() 306 auto module = CreateNewVerifiedModule(); in TEST_F() 335 auto module = CreateNewVerifiedModule(); in TEST_F() 364 auto module = CreateNewVerifiedModule(); in TEST_F() 418 auto module = CreateNewVerifiedModule(); in TEST_F() [all …]
|
D | defuser_test.cc | 46 auto m = CreateNewVerifiedModule(); in TEST_F() 62 auto m = CreateNewVerifiedModule(); in TEST_F() 86 auto m = CreateNewVerifiedModule(); in TEST_F() 112 auto m = CreateNewVerifiedModule(); in TEST_F() 151 auto m = CreateNewVerifiedModule(); in TEST_F() 191 auto m = CreateNewVerifiedModule(); in TEST_F()
|
D | hlo_constant_folding_test.cc | 50 auto module = CreateNewVerifiedModule(); in TEST_F() 72 auto module = CreateNewVerifiedModule(); in TEST_F() 94 auto module = CreateNewVerifiedModule(); in TEST_F() 137 auto module = CreateNewVerifiedModule(); in TEST_F() 164 auto module = CreateNewVerifiedModule(); in TEST_F() 189 auto module = CreateNewVerifiedModule(); in TEST_F()
|
D | call_graph_test.cc | 97 auto module = CreateNewVerifiedModule(); in TEST_F() 117 auto module = CreateNewVerifiedModule(); in TEST_F() 141 auto module = CreateNewVerifiedModule(); in TEST_F() 172 auto module = CreateNewVerifiedModule(); in TEST_F() 205 auto module = CreateNewVerifiedModule(); in TEST_F() 249 auto module = CreateNewVerifiedModule(); in TEST_F() 311 auto module = CreateNewVerifiedModule(); in TEST_F() 442 auto module = CreateNewVerifiedModule(); in TEST_F() 503 auto module = CreateNewVerifiedModule(); in TEST_F() 518 auto module = CreateNewVerifiedModule(); in TEST_F() [all …]
|
D | buffer_assignment_test.cc | 344 auto module = CreateNewVerifiedModule(); in TEST_F() 368 auto module = CreateNewVerifiedModule(); in TEST_F() 397 auto module = CreateNewVerifiedModule(); in TEST_F() 420 auto module = CreateNewVerifiedModule(); in TEST_F() 449 auto module = CreateNewVerifiedModule(); in TEST_F() 492 auto module = CreateNewVerifiedModule(); in TEST_F() 532 auto module = CreateNewVerifiedModule(); in TEST_F() 582 auto module = CreateNewVerifiedModule(); in TEST_F() 648 auto module = CreateNewVerifiedModule(); in TEST_F() 726 auto module = CreateNewVerifiedModule(); in TEST_F() [all …]
|
D | call_inliner_test.cc | 54 auto module = CreateNewVerifiedModule(); in TEST_F() 82 auto module = CreateNewVerifiedModule(); in TEST_F() 123 auto module = CreateNewVerifiedModule(); in TEST_F() 148 auto module = CreateNewVerifiedModule(); in TEST_F() 187 auto module = CreateNewVerifiedModule(); in TEST_F()
|
D | hlo_rematerialization_test.cc | 61 auto module = CreateNewVerifiedModule(); in TEST_F() 102 auto module = CreateNewVerifiedModule(); in TEST_F() 124 auto module = CreateNewVerifiedModule(); in TEST_F() 160 auto module = CreateNewVerifiedModule(); in TEST_F() 192 auto module = CreateNewVerifiedModule(); in TEST_F() 247 auto module = CreateNewVerifiedModule(); in TEST_F() 321 auto module = CreateNewVerifiedModule(); in TEST_F() 402 auto module = CreateNewVerifiedModule(); in TEST_F() 480 auto module = CreateNewVerifiedModule(); in TEST_P()
|
D | conditional_simplifier_test.cc | 103 auto m = CreateNewVerifiedModule(); in TEST_F() 111 auto m = CreateNewVerifiedModule(); in TEST_F() 121 auto m = CreateNewVerifiedModule(); in TEST_F() 133 auto m = CreateNewVerifiedModule(); in TEST_F() 149 auto m = CreateNewVerifiedModule(); in TEST_F() 163 auto m = CreateNewVerifiedModule(); in TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | gpu_conv_rewriter_test.cc | 123 auto module = CreateNewVerifiedModule(); in TEST_F() 177 auto module = CreateNewVerifiedModule(); in TEST_F() 239 auto module = CreateNewVerifiedModule(); in TEST_F() 269 auto module = CreateNewVerifiedModule(); in TEST_F() 299 auto module = CreateNewVerifiedModule(); in TEST_F() 328 auto module = CreateNewVerifiedModule(); in TEST_F() 381 auto module = CreateNewVerifiedModule(); in TEST_F() 429 auto module = CreateNewVerifiedModule(); in TEST_F() 463 auto module = CreateNewVerifiedModule(); in TEST_F() 517 auto module = CreateNewVerifiedModule(); in TEST_F() [all …]
|
D | instruction_fusion_test.cc | 45 auto module = CreateNewVerifiedModule(); in TEST_F() 65 auto module = CreateNewVerifiedModule(); in TEST_F() 84 auto module = CreateNewVerifiedModule(); in TEST_F() 103 auto module = CreateNewVerifiedModule(); in TEST_F() 121 auto module = CreateNewVerifiedModule(); in TEST_F() 138 auto module = CreateNewVerifiedModule(); in TEST_F() 482 auto module = CreateNewVerifiedModule(); in TEST_F() 585 auto m = CreateNewVerifiedModule(); in TEST_F() 603 auto m = CreateNewVerifiedModule(); in TEST_F() 621 auto m = CreateNewVerifiedModule(); in TEST_F()
|
D | stream_assignment_test.cc | 33 std::unique_ptr<HloModule> CreateNewVerifiedModule() { in CreateNewVerifiedModule() function in xla::gpu::StreamAssignmentTest 58 auto module = CreateNewVerifiedModule(); in TEST_F() 79 auto module = CreateNewVerifiedModule(); in TEST_F() 123 auto module = CreateNewVerifiedModule(); in TEST_F()
|