| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | gpu_kernel_tiling_test.cc | 78 CompileAndVerifyIr(std::move(hlo_module), expected_ir, in TEST_F()
110 CompileAndVerifyIr(std::move(hlo_module), expected_ir, in TEST_F()
159 CompileAndVerifyIr(std::move(hlo_module), expected_ir, in TEST_F()
199 CompileAndVerifyIr(std::move(hlo_module), expected_ir, in TEST_F()
240 CompileAndVerifyIr(std::move(hlo_module), expected_ir, in TEST_F()
273 CompileAndVerifyIr(std::move(hlo_module), expected_ir, in TEST_F()
306 CompileAndVerifyIr(std::move(hlo_module), expected_ir, in TEST_F()
347 CompileAndVerifyIr(std::move(hlo_module), expected_ir, in TEST_F()
398 CompileAndVerifyIr(std::move(hlo_module), expected_ir, in TEST_F()
456 CompileAndVerifyIr(std::move(hlo_module), expected_ir, in TEST_F()
[all …]
|
| D | gpu_unrolling_test.cc | 54 CompileAndVerifyIr(std::move(hlo_module), in TEST_F()
74 CompileAndVerifyIr(std::move(hlo_module), in TEST_F()
94 CompileAndVerifyIr(std::move(hlo_module), in TEST_F()
126 CompileAndVerifyIr(std::move(hlo_module), in TEST_F()
171 CompileAndVerifyIr(std::move(hlo_module), expected_ir, in TEST_F()
205 CompileAndVerifyIr(std::move(hlo_module), expected_ir, in TEST_F()
225 CompileAndVerifyIr(std::move(hlo_module), in TEST_F()
250 CompileAndVerifyIr(std::move(hlo_module), in TEST_F()
289 CompileAndVerifyIr(std::move(hlo_module), in TEST_F()
|
| D | gpu_index_test.cc | 59 CompileAndVerifyIr(std::move(hlo_module), in TEST_F()
83 CompileAndVerifyIr(std::move(module), in TEST_F()
107 CompileAndVerifyIr(std::move(module), in TEST_F()
145 CompileAndVerifyIr(std::move(module), in TEST_F()
172 CompileAndVerifyIr(std::move(module), in TEST_F()
202 CompileAndVerifyIr(std::move(module), in TEST_F()
|
| D | gpu_atomic_test.cc | 51 CompileAndVerifyIr(hlo_string, R"( in TEST_F()
78 CompileAndVerifyIr(hlo_string, R"( in TEST_F()
106 CompileAndVerifyIr(hlo_string, R"( in TEST_F()
144 CompileAndVerifyIr(hlo_string, R"( in TEST_F()
|
| D | gpu_input_fusible_slice_test.cc | 76 CompileAndVerifyIr(std::move(hlo_module), expected_ir, in TEST_F()
118 CompileAndVerifyIr(std::move(hlo_module), expected_ir, in TEST_F()
165 CompileAndVerifyIr(std::move(hlo_module), expected_ir, in TEST_F()
|
| D | parallel_reduction_test.cc | 64 CompileAndVerifyIr(std::move(hlo_module), in TEST_F()
177 CompileAndVerifyIr(std::move(hlo_module), in TEST_F()
|
| D | gpu_alignment_test.cc | 47 CompileAndVerifyIr(hlo_string, R"( in TEST_F()
|
| D | gpu_dyn_shape_test.cc | 38 CompileAndVerifyIr(std::move(hlo_module), in TEST_F()
|
| D | gpu_copy_test.cc | 52 CompileAndVerifyIr(std::move(hlo_module), "; CHECK-NOT: define void @_copy", in TEST_F()
|
| D | gpu_noalias_test.cc | 53 CompileAndVerifyIr( in TEST_F()
|
| D | gpu_fusion_test.cc | 51 CompileAndVerifyIr(hlo_text, in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | llvm_irgen_test_base.cc | 49 void LlvmIrGenTestBase::CompileAndVerifyIr( in CompileAndVerifyIr() function in xla::LlvmIrGenTestBase
62 void LlvmIrGenTestBase::CompileAndVerifyIr(const string& hlo_text, in CompileAndVerifyIr() function in xla::LlvmIrGenTestBase
69 CompileAndVerifyIr(std::move(module), expected_llvm_ir, match_optimized_ir); in CompileAndVerifyIr()
|
| D | llvm_irgen_test_base.h | 38 void CompileAndVerifyIr(std::unique_ptr<HloModule> hlo_module,
43 void CompileAndVerifyIr(const string& hlo_text,
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | alias_analysis_test.cc | 59 CompileAndVerifyIr(hlo_string, R"( in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_external_constants_test.cc | 52 CompileAndVerifyIr(std::move(module), filecheck_pattern, in TestWithArray()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | horizontal_input_fusion_test.cc | 137 CompileAndVerifyIr(module->Clone(), R"(CHECK: reduce-group-6)", in TEST_F()
|