| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | indexed_array_analysis_test.cc | 25 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 …]
|
| D | batch_dot_simplification_test.cc | 30 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 …]
|
| D | elemental_ir_emitter_test.cc | 31 void RunTest(const string& hlo_text, absl::Span<Literal* const> args) { in RunTest() argument
35 ParseAndReturnVerifiedModule(hlo_text, config)); in RunTest()
41 const string hlo_text = R"( in XLA_TEST_F() local
61 RunTest(hlo_text, {&lhs, &rhs}); in XLA_TEST_F()
65 const char* hlo_text = R"( in XLA_TEST_F() local
93 ParseAndReturnVerifiedModule(hlo_text, config)); in XLA_TEST_F()
|
| D | hlo_evaluator_test.cc | 2089 const char* hlo_text = R"( in TEST_F() local
2103 TF_ASSERT_OK_AND_ASSIGN(m_, ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
2113 const char* hlo_text = R"( in TEST_F() local
2127 TF_ASSERT_OK_AND_ASSIGN(m_, ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
2137 const char* hlo_text = R"( in TEST_F() local
2151 TF_ASSERT_OK_AND_ASSIGN(m_, ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
2163 const char* hlo_text = R"( in TEST_F() local
2177 TF_ASSERT_OK_AND_ASSIGN(m_, ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
2190 const char* hlo_text = R"( in TEST_F() local
2204 TF_ASSERT_OK_AND_ASSIGN(m_, ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
[all …]
|
| D | gather_expander_test.cc | 24 const string hlo_text = R"( in TEST() local
39 ParseHloString(hlo_text)); in TEST()
51 const string hlo_text = R"( in TEST() local
66 ParseHloString(hlo_text)); in TEST()
109 const string hlo_text = R"( in TEST() local
124 ParseHloString(hlo_text)); in TEST()
|
| D | buffer_assignment_test.cc | 1799 const char* hlo_text = R"( in TEST_F() local
1821 TF_ASSERT_OK_AND_ASSIGN(auto m, ParseAndReturnVerifiedModule(hlo_text)); in TEST_F()
2326 const char* hlo_text = R"( in TEST_F() local
2347 ParseAndReturnVerifiedModule(hlo_text, config)); in TEST_F()
|
| D | tuple_points_to_analysis_test.cc | 1030 const char* hlo_text = R"( in TEST_F() local
1050 TF_ASSERT_OK_AND_ASSIGN(module_, ParseHloString(hlo_text)); in TEST_F()
|
| D | hlo_dataflow_analysis_test.cc | 2409 const char* hlo_text = R"( in TEST_F() local
2429 TF_ASSERT_OK_AND_ASSIGN(module_, ParseHloString(hlo_text)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | gather_operation_test.cc | 32 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 ParseHloString(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()
68 const string hlo_text = R"( in XLA_TEST_F() local
85 RunTest(hlo_text, &operand, &start_indices); in XLA_TEST_F()
89 const string hlo_text = R"( in XLA_TEST_F() local
106 RunTest(hlo_text, &operand, &start_indices); in XLA_TEST_F()
[all …]
|
| D | scatter_test.cc | 30 void RunTest(const string& hlo_text, Literal* operand, in RunTest() argument
32 RunTest(hlo_text, {operand, scatter_indices, updates}); in RunTest()
35 void RunTest(const string& hlo_text, absl::Span<Literal* const> args) { in RunTest() argument
39 ParseHloString(hlo_text, config)); in RunTest()
45 const string hlo_text = R"( in XLA_TEST_F() local
69 RunTest(hlo_text, &operand, &scatter_indices, &updates); in XLA_TEST_F()
73 const string hlo_text = R"( in XLA_TEST_F() local
100 RunTest(hlo_text, &operand, &scatter_indices, &updates); in XLA_TEST_F()
104 const char* hlo_text = R"( in XLA_TEST_F() local
129 RunTest(hlo_text, &operand, &scatter_indices, &updates); in XLA_TEST_F()
[all …]
|
| D | llvm_irgen_test_base.cc | 63 void LlvmIrGenTestBase::CompileAndVerifyIr(const string& hlo_text, in CompileAndVerifyIr() argument
69 ParseHloString(hlo_text, config)); in CompileAndVerifyIr()
|
| D | conv_depthwise_backprop_filter_test.cc | 158 const string hlo_text = in XLA_TEST_P() local
162 hlo_text, ErrorSpec{0.01, 0.01}, [](HloModule* module) -> Status { in XLA_TEST_P()
|
| D | llvm_irgen_test_base.h | 43 void CompileAndVerifyIr(const string& hlo_text,
|
| D | conv_depthwise_test.cc | 215 const string hlo_text = in XLA_TEST_P() local
218 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{0.01, 0.01}, in XLA_TEST_P()
|
| D | grouped_convolution_test.cc | 227 const string hlo_text = BuildHloTextGroupedConvolution2D(spec, use_bfloat16); in XLA_TEST_P() local
229 EXPECT_TRUE(RunAndCompare(hlo_text, ErrorSpec{0.01, 0.01}, in XLA_TEST_P()
|
| D | hlo_test_base.cc | 147 HloTestBase::ParseAndReturnVerifiedModule(absl::string_view hlo_text, in ParseAndReturnVerifiedModule() argument
153 TF_RETURN_IF_ERROR(ParseHloString(hlo_text, module.get())); in ParseAndReturnVerifiedModule()
|
| D | hlo_test_base.h | 117 absl::string_view hlo_text,
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_outfeed_test.cc | 26 const string hlo_text = R"( in TEST_F() local
45 ParseHloString(hlo_text)); in TEST_F()
57 const string hlo_text = R"( in TEST_F() local
73 ParseHloString(hlo_text)); in TEST_F()
|
| D | cpu_literal_caching_test.cc | 28 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()
|
| D | cpu_key_value_sort_test.cc | 26 const string hlo_text = R"( in TEST_F() local
47 ParseHloString(hlo_text)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | gpu_fusion_test.cc | 31 const char* hlo_text = R"( in TEST_F() local
49 CompileAndVerifyIr(hlo_text, in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment_test.cc | 340 const char* hlo_text = R"( in TEST_F() local
351 ParseHloString(hlo_text)); in TEST_F()
369 const char* hlo_text = R"( in TEST_F() local
389 ParseHloString(hlo_text)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | interactive_graphviz.cc | 70 string hlo_text; member
681 } else if (!opts.hlo_text.empty()) { in RealMain()
683 opts.hlo_text, xla::GetDebugOptionsFromFlags()) in RealMain()
723 tensorflow::Flag("hlo_text", &opts.hlo_text, in main()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client_test.py | 112 hlo_text = computation.GetHloText()
113 self.assertTrue(hlo_text.startswith("HloModule acomputation"))
|