Home
last modified time | relevance | path

Searched refs:hlo_text (Results 1 – 25 of 39) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dindexed_array_analysis_test.cc25 void AssertArrayForRootExpressionIs(const string& hlo_text, in AssertArrayForRootExpressionIs() argument
27 AssertArrayForRootExpressionIsImpl(hlo_text, root_expression, in AssertArrayForRootExpressionIs()
32 const string& hlo_text, const string& root_expression) { in AssertArrayWithConstantsForRootExpressionIs() argument
33 AssertArrayForRootExpressionIsImpl(hlo_text, root_expression, in AssertArrayWithConstantsForRootExpressionIs()
59 void AssertArrayForRootExpressionIsImpl(const string& hlo_text, in AssertArrayForRootExpressionIsImpl() argument
64 ParseAndReturnVerifiedModule(hlo_text)); in AssertArrayForRootExpressionIsImpl()
77 string hlo_text = R"( in TEST_F() local
92 AssertArrayForRootExpressionIs(hlo_text, in TEST_F()
97 string hlo_text = R"( in TEST_F() local
113 hlo_text, "(scalar-indexed-const (constant s32[3,3]) %indices 0->[0])"); in TEST_F()
[all …]
Delemental_ir_emitter_test.cc31 void RunTest(const string& hlo_text, absl::Span<Literal* const> args) { in RunTest() argument
35 ParseAndReturnVerifiedModule(hlo_text, config)); in RunTest()
39 void RunTypeConversionTest(absl::string_view hlo_text) { in RunTypeConversionTest() argument
46 ParseAndReturnVerifiedModule(hlo_text, config)); in RunTypeConversionTest()
52 const string hlo_text = R"( in XLA_TEST_F() local
72 RunTest(hlo_text, {&lhs, &rhs}); in XLA_TEST_F()
76 const char* hlo_text = R"( in XLA_TEST_F() local
96 RunTest(hlo_text, {&lhs, &rhs}); in XLA_TEST_F()
100 const char* hlo_text = R"( in XLA_TEST_F() local
128 ParseAndReturnVerifiedModule(hlo_text, config)); in XLA_TEST_F()
[all …]
Dbatch_dot_simplification_test.cc30 const string hlo_text = R"( in TEST_F() local
41 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
54 const string hlo_text = R"( in TEST_F() local
65 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
78 const string hlo_text = R"( in TEST_F() local
89 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
102 const string hlo_text = R"( in TEST_F() local
113 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
126 const string hlo_text = R"( in TEST_F() local
137 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
[all …]
Ddynamic_padder_test.cc224 std::unique_ptr<HloModule> GetHloModule(const string& hlo_text) { in GetHloModule() argument
226 ParseAndReturnVerifiedModule(hlo_text).ValueOrDie(); in GetHloModule()
244 const string hlo_text = R"( in XLA_TEST_F() local
267 absl::StrReplaceAll(hlo_text, {{"INDICES_BOUND", "2"}}); in XLA_TEST_F()
282 absl::StrReplaceAll(hlo_text, {{"INDICES_BOUND", "4"}}); in XLA_TEST_F()
307 const string hlo_text = R"( in XLA_TEST_F() local
330 auto module_not_padded = GetHloModule(hlo_text); in XLA_TEST_F()
340 auto module_padded = GetHloModule(hlo_text); in XLA_TEST_F()
373 const string hlo_text = R"( in XLA_TEST_F() local
402 auto module = GetHloModule(hlo_text); in XLA_TEST_F()
[all …]
Dhlo_evaluator_test.cc1584 const char* hlo_text = R"( variable
1596 TF_ASSERT_OK_AND_ASSIGN(m_, ParseAndReturnVerifiedModule(hlo_text));
1603 const char* hlo_text = R"( variable
1615 TF_ASSERT_OK_AND_ASSIGN(m_, ParseAndReturnVerifiedModule(hlo_text));
1622 const char* hlo_text = R"( variable
1633 TF_ASSERT_OK_AND_ASSIGN(m_, ParseAndReturnVerifiedModule(hlo_text));
1640 const char* hlo_text = R"( variable
1651 TF_ASSERT_OK_AND_ASSIGN(m_, ParseAndReturnVerifiedModule(hlo_text));
1660 const char* hlo_text = R"( variable
1668 TF_ASSERT_OK_AND_ASSIGN(m_, ParseAndReturnVerifiedModule(hlo_text));
[all …]
Dconditional_to_select_test.cc40 const string hlo_text = R"( in TEST_F() local
68 auto module = ParseAndReturnVerifiedModule(hlo_text).ValueOrDie(); in TEST_F()
83 const string hlo_text = R"( in TEST_F() local
121 auto module = ParseAndReturnVerifiedModule(hlo_text).ValueOrDie(); in TEST_F()
138 const string hlo_text = R"( in TEST_F() local
177 auto module = ParseAndReturnVerifiedModule(hlo_text).ValueOrDie(); in TEST_F()
Dgather_expander_test.cc28 const string hlo_text = R"( in TEST_F() local
43 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
55 const string hlo_text = R"( in TEST_F() local
70 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
113 const string hlo_text = R"( in TEST_F() local
128 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dtree_reduction_rewriter_test.cc43 void EnsureDeterminism(absl::string_view hlo_text) { in EnsureDeterminism() argument
47 EXPECT_TRUE(RunMultipleTimes(hlo_text, in EnsureDeterminism()
56 const char* hlo_text = R"( in TEST_F() local
74 MatchOptimizedHloWithShapes(hlo_text, in TEST_F()
84 EnsureDeterminism(hlo_text); in TEST_F()
85 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F()
89 const char* hlo_text = R"( in TEST_F() local
106 EnsureDeterminism(hlo_text); in TEST_F()
108 MatchOptimizedHloWithShapes(hlo_text, in TEST_F()
122 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F()
[all …]
Dreduction_dimension_grouper_test.cc42 const char* hlo_text = R"( in TEST_F() local
61 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F()
62 MatchOptimizedHloWithShapes(hlo_text, in TEST_F()
Dreduction_layout_normalizer_test.cc43 const char* hlo_text = R"( in TEST_F() local
61 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F()
62 MatchOptimizedHloWithShapes(hlo_text, in TEST_F()
Dreduction_degenerate_dim_remover_test.cc46 const char* hlo_text = R"( in TEST_F() local
64 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F()
65 MatchOptimizedHloWithShapes(hlo_text, in TEST_F()
Dgpu_fusion_test.cc31 const char* hlo_text = R"( in TEST_F() local
49 CompileAndVerifyIr(hlo_text, in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dgather_operation_test.cc31 void RunTest(const string& hlo_text, Literal* operand, in RunTest() argument
33 RunTest(hlo_text, {operand, start_indices}); in RunTest()
36 void RunTest(const string& hlo_text, absl::Span<Literal* const> args) { in RunTest() argument
40 ParseAndReturnVerifiedModule(hlo_text, config)); in RunTest()
46 const string hlo_text = R"( in XLA_TEST_F() local
63 RunTest(hlo_text, &operand, &start_indices); in XLA_TEST_F()
67 const string hlo_text = R"( in XLA_TEST_F() local
84 RunTest(hlo_text, &operand, &start_indices); in XLA_TEST_F()
88 const string hlo_text = R"( in XLA_TEST_F() local
105 RunTest(hlo_text, &operand, &start_indices); in XLA_TEST_F()
[all …]
Dscatter_test.cc29 void RunTest(const string& hlo_text, Literal* operand, in RunTest() argument
31 RunTest(hlo_text, {operand, scatter_indices, updates}); in RunTest()
34 void RunTest(const string& hlo_text, absl::Span<Literal* const> args) { in RunTest() argument
38 ParseAndReturnVerifiedModule(hlo_text, config)); in RunTest()
44 const string hlo_text = R"( in XLA_TEST_F() local
68 RunTest(hlo_text, &operand, &scatter_indices, &updates); in XLA_TEST_F()
72 const string hlo_text = R"( in XLA_TEST_F() local
99 RunTest(hlo_text, &operand, &scatter_indices, &updates); in XLA_TEST_F()
103 const char* hlo_text = R"( in XLA_TEST_F() local
128 RunTest(hlo_text, &operand, &scatter_indices, &updates); in XLA_TEST_F()
[all …]
Dlocal_client_test_base.cc216 LocalClientTestBase::ParseAndReturnVerifiedModule(absl::string_view hlo_text) { in ParseAndReturnVerifiedModule() argument
217 return ParseAndReturnVerifiedModule(hlo_text, HloModuleConfig()); in ParseAndReturnVerifiedModule()
222 absl::string_view hlo_text, const HloModuleConfig& config) { in ParseAndReturnVerifiedModule() argument
227 TF_RETURN_IF_ERROR(module->ParseHloStringAndVerifyModule(hlo_text)); in ParseAndReturnVerifiedModule()
Dconv_depthwise_test.cc96 const string hlo_text = in XLA_TEST_P() local
99 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{0.01, 0.01}, in XLA_TEST_P()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_outfeed_test.cc28 const string hlo_text = R"( in TEST_F() local
46 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
58 const string hlo_text = R"( in TEST_F() local
73 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
Dcpu_literal_caching_test.cc28 const string hlo_text = R"( in TEST_F() local
64 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
78 const string hlo_text = R"( in TEST_F() local
112 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
Dtree_reduction_rewriter_test.cc38 const char* hlo_text = R"( in TEST_F() local
54 MatchOptimizedHlo(hlo_text, in TEST_F()
Dcpu_key_value_sort_test.cc28 const string hlo_text = R"( in TEST_F() local
48 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
Dcpu_profiling_test.cc78 constexpr char hlo_text[] = R"(HloModule test in TEST_P() local
88 auto hlo_module = ParseAndReturnVerifiedModule(hlo_text, config); in TEST_P()
/external/tensorflow/tensorflow/compiler/xla/service/mlir_gpu/experimental/conv_emitter/
Dconv_emitter_test.cc38 std::string CompileHloConvAndGetMlir(absl::string_view hlo_text) { in CompileHloConvAndGetMlir() argument
44 TF_CHECK_OK(hlo_module.ParseHloStringAndVerifyModule(hlo_text)); in CompileHloConvAndGetMlir()
79 std::string hlo_text = R"(HloModule TestModule in TEST() local
136 RunFileCheck(CompileHloConvAndGetMlir(hlo_text), expected_mlir_pattern) in TEST()
/external/tensorflow/tensorflow/compiler/xla/service/mlir_gpu/
Dmlir_irgen_test_base.cc79 void MlirIrGenTestBase::CompileAndVerifyIr(const string& hlo_text, in CompileAndVerifyIr() argument
88 TF_ASSERT_OK(module->ParseHloStringAndVerifyModule(hlo_text)); in CompileAndVerifyIr()
130 void MlirIrGenTestBase::CompileAndVerifyErrors(const string& hlo_text, in CompileAndVerifyErrors() argument
139 TF_ASSERT_OK(module->ParseHloStringAndVerifyModule(hlo_text)); in CompileAndVerifyErrors()
Dmlir_irgen_test_base.h46 void CompileAndVerifyIr(const string& hlo_text,
64 void CompileAndVerifyErrors(const string& hlo_text,
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment_test.cc342 const char* hlo_text = R"( in TEST_F() local
353 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
371 const char* hlo_text = R"( in TEST_F() local
391 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
408 const char* hlo_text = R"( in TEST_F() local
418 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()

12