Home
last modified time | relevance | path

Searched refs:hlo_text (Results 1 – 25 of 44) 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 …]
Ddynamic_padder_test.cc79 std::unique_ptr<HloModule> GetHloModule(const string& hlo_text) { in GetHloModule() argument
81 ParseAndReturnVerifiedModule(hlo_text).ValueOrDie(); in GetHloModule()
145 const string hlo_text = R"( in TEST_F() local
162 module_ = GetHloModule(hlo_text); in TEST_F()
198 const string hlo_text = R"( in TEST_F() local
218 module_ = GetHloModule(hlo_text); in TEST_F()
374 std::unique_ptr<HloModule> GetHloModule(const string& hlo_text) { in GetHloModule() argument
376 ParseAndReturnVerifiedModule(hlo_text).ValueOrDie(); in GetHloModule()
400 const string hlo_text = R"( in XLA_TEST_F() local
423 absl::StrReplaceAll(hlo_text, {{"INDICES_BOUND", "2"}}); in XLA_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 …]
Dhlo_evaluator_test.cc1585 const char* hlo_text = R"( variable
1597 TF_ASSERT_OK_AND_ASSIGN(m_, ParseAndReturnVerifiedModule(hlo_text));
1604 const char* hlo_text = R"( variable
1616 TF_ASSERT_OK_AND_ASSIGN(m_, ParseAndReturnVerifiedModule(hlo_text));
1623 const char* hlo_text = R"( variable
1634 TF_ASSERT_OK_AND_ASSIGN(m_, ParseAndReturnVerifiedModule(hlo_text));
1641 const char* hlo_text = R"( variable
1652 TF_ASSERT_OK_AND_ASSIGN(m_, ParseAndReturnVerifiedModule(hlo_text));
1661 const char* hlo_text = R"( variable
1669 TF_ASSERT_OK_AND_ASSIGN(m_, ParseAndReturnVerifiedModule(hlo_text));
[all …]
Dgather_expander_test.cc29 const string hlo_text = R"( in TEST_F() local
44 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
58 const string hlo_text = R"( in TEST_F() local
73 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
118 const string hlo_text = R"( in TEST_F() local
133 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
159 const string hlo_text = R"( in TEST_F() local
175 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
182 const string hlo_text = R"( in TEST_F() local
198 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
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()
/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
73 MatchOptimizedHloWithShapes(hlo_text, in TEST_F()
90 EnsureDeterminism(hlo_text); in TEST_F()
91 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F()
95 const char* hlo_text = R"( in TEST_F() local
111 EnsureDeterminism(hlo_text); in TEST_F()
113 MatchOptimizedHloWithShapes(hlo_text, in TEST_F()
130 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F()
[all …]
Dreduction_vectorization_test.cc38 const char* hlo_text = R"( in TEST_F() local
54 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
82 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F()
86 const char* hlo_text = R"( in TEST_F() local
102 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
130 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F()
134 const char* hlo_text = R"( in TEST_F() local
150 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
183 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F()
187 const char* hlo_text = R"( in TEST_F() local
[all …]
Dreduction_degenerate_dim_remover_test.cc47 const char* hlo_text = R"( in TEST_F() local
65 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F()
66 MatchOptimizedHloWithShapes(hlo_text, in TEST_F()
73 const char* hlo_text = R"( in TEST_F() local
91 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F()
95 MatchOptimizedHloWithShapes(hlo_text, in TEST_F()
Dparallel_reduction_test.cc36 const char* hlo_text = R"( in TEST_F() local
63 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
71 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F()
133 const char* hlo_text = R"( in TEST_F() local
176 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
185 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F()
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.cc44 const char* hlo_text = R"( in TEST_F() local
62 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F()
63 MatchOptimizedHloWithShapes(hlo_text, in TEST_F()
Dgpu_copy_alone_test.cc41 const char* hlo_text = R"( in TEST_F() local
49 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
51 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F()
Dsorting_test.cc49 const char* hlo_text = R"( in TEST_F() local
66 EXPECT_TRUE(RunAndCompareNoHloPasses(hlo_text, ErrorSpec{1e-5, 1e-5})); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dgather_operation_test.cc32 void RunTest(const string& hlo_text, Literal* operand, in RunTest() argument
34 RunTest(hlo_text, {operand, start_indices}); in RunTest()
37 void RunTest(const string& hlo_text, absl::Span<Literal* const> args) { in RunTest() argument
41 ParseAndReturnVerifiedModule(hlo_text, config)); in RunTest()
47 const string hlo_text = R"( in XLA_TEST_F() local
64 RunTest(hlo_text, &operand, &start_indices); in XLA_TEST_F()
69 const string hlo_text = R"( in XLA_TEST_F() local
88 RunTest(hlo_text, &operand, &start_indices); in XLA_TEST_F()
92 const string hlo_text = R"( in XLA_TEST_F() local
109 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 …]
Diota_test.cc30 const string hlo_text = R"( in XLA_TEST_F() local
37 EXPECT_TRUE(RunAndCompare(hlo_text, absl::nullopt)); in XLA_TEST_F()
41 const string hlo_text = R"( in XLA_TEST_F() local
48 EXPECT_TRUE(RunAndCompare(hlo_text, absl::nullopt)); in XLA_TEST_F()
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()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_outfeed_test.cc29 const string hlo_text = R"( in TEST_F() local
47 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
60 const string hlo_text = R"( in TEST_F() local
75 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
88 const string hlo_text = R"( in TEST_F() local
103 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
Dcpu_literal_caching_test.cc29 const string hlo_text = R"( in TEST_F() local
65 ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
80 const string hlo_text = R"( in TEST_F() local
114 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.cc29 const string hlo_text = R"( in TEST_F() local
49 TF_ASSERT_OK_AND_ASSIGN(auto module, ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/
Dconv_emitter_test.cc40 std::string CompileHloConvAndGetMlir(absl::string_view hlo_text) { in CompileHloConvAndGetMlir() argument
46 TF_CHECK_OK(hlo_module.ParseHloStringAndVerifyModule(hlo_text)); in CompileHloConvAndGetMlir()
81 std::string hlo_text = R"(HloModule TestModule in TEST() local
138 RunFileCheck(CompileHloConvAndGetMlir(hlo_text), expected_mlir_pattern) in TEST()
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client_backend_independent_test.py67 hlo_text = computation.as_hlo_text()
68 self.assertTrue(hlo_text.startswith("HloModule acomputation"))
77 hlo_text = computation.as_hlo_module().to_string()
78 self.assertTrue(hlo_text.startswith("HloModule acomputation"))

12