| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | hlo_test_base.cc | 107 HloTestBase::HloTestBase(bool verifier_layout_sensitive, in HloTestBase() function in xla::HloTestBase
111 : HloTestBase(GetTestPlatform(), GetReferencePlatform(), in HloTestBase()
116 HloTestBase::HloTestBase(se::Platform* test_platform, in HloTestBase() function in xla::HloTestBase
133 std::unique_ptr<HloModule> HloTestBase::CreateNewUnverifiedModule( in CreateNewUnverifiedModule()
138 std::unique_ptr<VerifiedHloModule> HloTestBase::CreateNewVerifiedModule( in CreateNewVerifiedModule()
147 HloTestBase::ParseAndReturnVerifiedModule(absl::string_view hlo_text, in ParseAndReturnVerifiedModule()
159 StatusOr<bool> HloTestBase::RunHloPass(HloPassInterface* hlo_pass, in RunHloPass()
174 PrecisionConfig HloTestBase::DefaultPrecisionConfig(int operands) { in DefaultPrecisionConfig()
181 DebugOptions HloTestBase::GetDebugOptionsForTest() { in GetDebugOptionsForTest()
190 StatusOr<Literal> HloTestBase::Execute(std::unique_ptr<HloModule> module, in Execute()
[all …]
|
| D | sample_file_test.cc | 34 class SampleFileTest : public HloTestBase {
37 : HloTestBase( in SampleFileTest()
|
| D | hlo_test_base.h | 96 class HloTestBase : public ::testing::Test {
133 HloTestBase(bool verifier_layout_sensitive = false,
141 HloTestBase(se::Platform* test_platform, se::Platform* reference_platform,
147 ~HloTestBase() override {} in ~HloTestBase()
|
| D | ptxas_bug_120501638.cc | 24 class PtxasBugTest : public HloTestBase {};
|
| D | sample_text_test.cc | 35 class SampleTextTest : public HloTestBase {};
|
| D | codegen_test_base.h | 29 class CodegenTestBase : public HloTestBase {
|
| D | multi_device_all_reduce_test.cc | 27 class MultiDeviceAllReduceTest : public HloTestBase {};
|
| D | all_reduce_test.cc | 27 class TrivialCrossReplicaSumTest : public HloTestBase {};
|
| D | reduce_hlo_test.cc | 52 : public HloTestBase,
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | gpu_unrolling_test.cc | 48 auto debug_options = HloTestBase::GetDebugOptionsForTest(); in TEST_F()
65 auto debug_options = HloTestBase::GetDebugOptionsForTest(); in TEST_F()
108 auto debug_options = HloTestBase::GetDebugOptionsForTest(); in TEST_F()
139 auto debug_options = HloTestBase::GetDebugOptionsForTest(); in TEST_F()
|
| D | gpu_index_test.cc | 69 config.set_debug_options(HloTestBase::GetDebugOptionsForTest()); in TEST_F()
93 config.set_debug_options(HloTestBase::GetDebugOptionsForTest()); in TEST_F()
120 auto debug_options = HloTestBase::GetDebugOptionsForTest(); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_padder_test.cc | 41 class DynamicPadderTest : public HloTestBase {
43 DynamicPadderTest() : HloTestBase() { module_ = CreateNewVerifiedModule(); } in DynamicPadderTest()
131 HloTestBase::DefaultPrecisionConfig(2))); in TEST_F()
174 HloTestBase::DefaultPrecisionConfig(2))); in TEST_F()
|
| D | hlo_verifier_test.cc | 46 class HloVerifierTest : public HloTestBase {
49 : HloTestBase(/*verifier_layout_sensitive=*/false, in HloVerifierTest()
53 class HloVerifierTestAllowMixedPrecision : public HloTestBase {
56 : HloTestBase(/*verifier_layout_sensitive=*/false, in HloVerifierTestAllowMixedPrecision()
60 class HloVerifierTestLayoutSensitive : public HloTestBase {
63 : HloTestBase(/*verifier_layout_sensitive=*/true, in HloVerifierTestLayoutSensitive()
|
| D | zero_sized_hlo_elimination_test.cc | 38 class ZeroSizedHloEliminationTest : public HloTestBase {
41 : HloTestBase(), in ZeroSizedHloEliminationTest()
|
| D | hlo_get_dimension_size_rewriter_test.cc | 37 class HloGetDimensionSizeRewriterTest : public HloTestBase {
|
| D | while_loop_analysis_test.cc | 26 class WhileLoopAnalysisTest : public HloTestBase {};
|
| D | dfs_hlo_visitor_with_default_test.cc | 32 class DfsHloVisitorWithDefaultTest : public HloTestBase {};
|
| D | elemental_ir_emitter_test.cc | 29 class ElementalIrEmitterExecutionTest : public HloTestBase {
|
| D | hlo_execution_profile_test.cc | 29 class HloExecutionProfileTest : public HloTestBase {};
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | parallel_task_assignment_test.cc | 26 class ParallelTaskAssignmentTest : public HloTestBase {
38 : HloTestBase(), target_machine_features_([](int64 shape_size) { in ParallelTaskAssignmentTest()
|
| D | shape_partition_test.cc | 29 class ShapePartitionAssignerTest : public HloTestBase {
94 class ShapePartitionIteratorTest : public HloTestBase {
148 class RandomShapePartitionIteratorTest : public HloTestBase {
|
| D | cpu_hlo_support_checker_test.cc | 28 class CpuHloSupportCheckerTest : public HloTestBase {
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_sanitize_constant_names_test.cc | 31 using SanitizeConstantNamesTest = HloTestBase;
|
| D | variadic_op_splitter_test.cc | 35 class VariadicOpSplitterTest : public HloTestBase {};
|
| D | gpu_hlo_support_checker_test.cc | 28 class GpuHloSupportCheckerTest : public HloTestBase {
|