Searched refs:VerifiedHloModule (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_pass_pipeline_test.cc | 35 TF_ASSIGN_OR_RETURN(std::unique_ptr<VerifiedHloModule> module, in ParseModuleGroup()
109 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> module, in TEST_F()
132 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> module, in TEST_F()
194 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> module, in TEST_F()
246 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> module, in TEST_F()
|
| D | while_loop_simplifier_test.cc | 48 TF_MUST_USE_RESULT std::unique_ptr<VerifiedHloModule>
54 TF_MUST_USE_RESULT std::unique_ptr<VerifiedHloModule>
58 std::unique_ptr<VerifiedHloModule>
91 std::unique_ptr<VerifiedHloModule>
|
| D | layout_assignment_test.cc | 1149 std::unique_ptr<VerifiedHloModule> m, in TEST_F()
1165 std::unique_ptr<VerifiedHloModule> m, in TEST_F()
1192 std::unique_ptr<VerifiedHloModule> m, in TEST_F()
1224 std::unique_ptr<VerifiedHloModule> m, in TEST_F()
1251 std::unique_ptr<VerifiedHloModule> m, in TEST_F()
1286 std::unique_ptr<VerifiedHloModule> m, in TEST_F()
|
| D | hlo_creation_utils_test.cc | 30 std::unique_ptr<VerifiedHloModule> CreateModuleWithProgramShape( in CreateModuleWithProgramShape()
|
| D | buffer_liveness_test.cc | 614 std::unique_ptr<VerifiedHloModule> BuildModule( in BuildModule()
|
| D | algebraic_simplifier_test.cc | 4901 std::unique_ptr<VerifiedHloModule> module = CreateNewVerifiedModule(); in TEST_F()
4923 std::unique_ptr<VerifiedHloModule> module = CreateNewVerifiedModule(); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | hlo_test_base.h | 45 class VerifiedHloModule : public HloModule {
47 VerifiedHloModule(const string& name, const HloModuleConfig& config, in VerifiedHloModule() function
56 ~VerifiedHloModule() override { VerifyOrAddFailure("in destructor"); } in ~VerifiedHloModule()
112 std::unique_ptr<VerifiedHloModule> CreateNewVerifiedModule(
116 StatusOr<std::unique_ptr<VerifiedHloModule>> ParseAndReturnVerifiedModule(
|
| D | hlo_test_base.cc | 88 Status VerifiedHloModule::Verify() { in Verify()
96 void VerifiedHloModule::VerifyOrAddFailure(const string& message) { in VerifyOrAddFailure()
138 std::unique_ptr<VerifiedHloModule> HloTestBase::CreateNewVerifiedModule( in CreateNewVerifiedModule()
140 return absl::make_unique<VerifiedHloModule>( in CreateNewVerifiedModule()
146 StatusOr<std::unique_ptr<VerifiedHloModule>>
149 auto module = absl::make_unique<VerifiedHloModule>( in ParseAndReturnVerifiedModule()
|