HomeSort by relevance Sort by last modified time
    Searched refs:VerifiedHloModule (Results 1 - 8 of 8) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_pass_pipeline_test.cc 35 TF_ASSIGN_OR_RETURN(std::unique_ptr<VerifiedHloModule> module,
109 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> module,
132 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> module,
194 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> module,
246 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> module,
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>
hlo_creation_utils_test.cc 30 std::unique_ptr<VerifiedHloModule> CreateModuleWithProgramShape(
layout_assignment_test.cc     [all...]
buffer_liveness_test.cc 614 std::unique_ptr<VerifiedHloModule> BuildModule(
    [all...]
algebraic_simplifier_test.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/tests/
hlo_test_base.h 45 class VerifiedHloModule : public HloModule {
47 VerifiedHloModule(const string& name, const HloModuleConfig& config,
56 ~VerifiedHloModule() override { VerifyOrAddFailure("in destructor"); }
112 std::unique_ptr<VerifiedHloModule> CreateNewVerifiedModule(
115 // Parses the given string and returns module as a VerifiedHloModule.
116 StatusOr<std::unique_ptr<VerifiedHloModule>> ParseAndReturnVerifiedModule(
hlo_test_base.cc 88 Status VerifiedHloModule::Verify() {
96 void VerifiedHloModule::VerifyOrAddFailure(const string& message) {
138 std::unique_ptr<VerifiedHloModule> HloTestBase::CreateNewVerifiedModule(
140 return absl::make_unique<VerifiedHloModule>(
146 StatusOr<std::unique_ptr<VerifiedHloModule>>
149 auto module = absl::make_unique<VerifiedHloModule>(

Completed in 1817 milliseconds