| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | reduction_vectorization_test.cc | 53 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> optimized_module, in TEST_F()
101 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> optimized_module, in TEST_F()
149 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> optimized_module, in TEST_F()
202 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> optimized_module, in TEST_F()
233 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> optimized_module, in TEST_F()
283 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> optimized_module, in TEST_F()
337 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> optimized_module, in TEST_F()
|
| D | gpu_codegen_test.cc | 33 std::unique_ptr<VerifiedHloModule>
43 return absl::make_unique<VerifiedHloModule>( in CreateNewVerifiedModuleWithFTZ()
50 std::unique_ptr<VerifiedHloModule> hlo_module, absl::string_view pattern) { in CompileAndOptionallyVerifyPtx()
|
| D | gpu_codegen_test.h | 38 std::unique_ptr<VerifiedHloModule> CreateNewVerifiedModuleWithFTZ(bool ftz);
46 std::unique_ptr<VerifiedHloModule> hlo_module, absl::string_view pattern);
|
| D | parallel_reduction_test.cc | 62 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> hlo_module, in TEST_F()
75 std::unique_ptr<VerifiedHloModule> module = CreateNewVerifiedModule(); in TEST_F()
175 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> hlo_module, in TEST_F()
|
| D | gpu_ftz_test.cc | 34 std::unique_ptr<VerifiedHloModule> CreateBinaryOpModule(HloOpcode op) { in CreateBinaryOpModule()
51 std::unique_ptr<VerifiedHloModule> CreateUnaryOpModule(HloOpcode op) { in CreateUnaryOpModule()
|
| D | gpu_copy_alone_test.cc | 48 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> optimized_module, in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | verified_hlo_module.h | 33 class VerifiedHloModule : public HloModule {
35 VerifiedHloModule(const string& name, const HloModuleConfig& config, in VerifiedHloModule() function
44 ~VerifiedHloModule() override { VerifyOrAddFailure("in destructor"); } in ~VerifiedHloModule()
|
| D | verified_hlo_module.cc | 29 Status VerifiedHloModule::ParseHloStringAndVerifyModule(absl::string_view str) { in ParseHloStringAndVerifyModule()
36 void VerifiedHloModule::VerifyOrAddFailure(absl::string_view message) { in VerifyOrAddFailure()
47 Status VerifiedHloModule::Verify() { in Verify()
|
| D | local_client_test_base.cc | 215 StatusOr<std::unique_ptr<VerifiedHloModule>>
220 StatusOr<std::unique_ptr<VerifiedHloModule>>
223 auto module = absl::make_unique<VerifiedHloModule>( in ParseAndReturnVerifiedModule()
|
| D | local_client_test_base.h | 117 StatusOr<std::unique_ptr<VerifiedHloModule>> ParseAndReturnVerifiedModule(
119 StatusOr<std::unique_ptr<VerifiedHloModule>> ParseAndReturnVerifiedModule(
|
| D | hlo_test_base.h | 87 std::unique_ptr<VerifiedHloModule> CreateNewVerifiedModule(
91 StatusOr<std::unique_ptr<VerifiedHloModule>> ParseAndReturnVerifiedModule(
93 StatusOr<std::unique_ptr<VerifiedHloModule>> ParseAndReturnVerifiedModule(
|
| D | hlo_test_base.cc | 119 std::unique_ptr<VerifiedHloModule> HloTestBase::CreateNewVerifiedModule( in CreateNewVerifiedModule()
121 return absl::make_unique<VerifiedHloModule>( in CreateNewVerifiedModule()
127 StatusOr<std::unique_ptr<VerifiedHloModule>>
134 StatusOr<std::unique_ptr<VerifiedHloModule>>
137 auto module = absl::make_unique<VerifiedHloModule>( in ParseAndReturnVerifiedModule()
|
| D | buffer_donation_test.cc | 289 StatusOr<std::unique_ptr<VerifiedHloModule>> module = in TEST_F()
317 StatusOr<std::unique_ptr<VerifiedHloModule>> module = in TEST_F()
|
| D | llvm_compiler_test.cc | 167 return absl::make_unique<VerifiedHloModule>( in CreateNewVerifiedModule()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_pass_pipeline_test.cc | 39 TF_ASSIGN_OR_RETURN(std::unique_ptr<VerifiedHloModule> module, in ParseModuleGroup()
113 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> module, in TEST_F()
136 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> module, in TEST_F()
198 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> module, in TEST_F()
250 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> module, in TEST_F()
|
| D | tuple_util_test.cc | 33 StatusOr<std::unique_ptr<VerifiedHloModule>> GetParsedModule( in GetParsedModule()
45 auto module = absl::make_unique<VerifiedHloModule>( in GetParsedModule()
|
| 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 | hlo_parser.h | 96 friend class VerifiedHloModule; variable
|
| D | hlo_creation_utils_test.cc | 30 std::unique_ptr<VerifiedHloModule> CreateModuleWithProgramShape( in CreateModuleWithProgramShape()
45 std::unique_ptr<VerifiedHloModule> CreateModuleWithProgramShape( in CreateModuleWithProgramShape()
|
| D | while_util_test.cc | 34 StatusOr<std::unique_ptr<VerifiedHloModule>> GetParsedModule( in GetParsedModule()
|
| D | layout_assignment_test.cc | 1143 std::unique_ptr<VerifiedHloModule> m, in TEST_F()
1159 std::unique_ptr<VerifiedHloModule> m, in TEST_F()
1186 std::unique_ptr<VerifiedHloModule> m, in TEST_F()
1218 std::unique_ptr<VerifiedHloModule> m, in TEST_F()
1245 std::unique_ptr<VerifiedHloModule> m, in TEST_F()
1280 std::unique_ptr<VerifiedHloModule> m, in TEST_F()
|
| D | conditional_simplifier_test.cc | 509 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> module, in TEST_F()
|
| D | algebraic_simplifier_test.cc | 5885 std::unique_ptr<VerifiedHloModule> module = CreateNewVerifiedModule(); in TEST_F()
5907 std::unique_ptr<VerifiedHloModule> module = CreateNewVerifiedModule(); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/ |
| D | conv_emitter_test.cc | 42 VerifiedHloModule hlo_module( in CompileHloConvAndGetMlir()
|