| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | verified_hlo_module.h | 33 class VerifiedHloModule : public HloModule {
35 VerifiedHloModule(const std::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 | 212 StatusOr<std::unique_ptr<VerifiedHloModule>>
217 StatusOr<std::unique_ptr<VerifiedHloModule>>
220 auto module = std::make_unique<VerifiedHloModule>( in ParseAndReturnVerifiedModule()
|
| D | local_client_test_base.h | 116 StatusOr<std::unique_ptr<VerifiedHloModule>> ParseAndReturnVerifiedModule(
118 StatusOr<std::unique_ptr<VerifiedHloModule>> ParseAndReturnVerifiedModule(
|
| D | hlo_test_base.h | 87 std::unique_ptr<VerifiedHloModule> CreateNewVerifiedModule(
91 StatusOr<std::unique_ptr<VerifiedHloModule>> ParseAndReturnVerifiedModule(
94 StatusOr<std::unique_ptr<VerifiedHloModule>> ParseAndReturnVerifiedModule(
|
| D | hlo_test_base.cc | 117 std::unique_ptr<VerifiedHloModule> HloTestBase::CreateNewVerifiedModule( in CreateNewVerifiedModule()
119 return std::make_unique<VerifiedHloModule>( in CreateNewVerifiedModule()
125 StatusOr<std::unique_ptr<VerifiedHloModule>>
137 StatusOr<std::unique_ptr<VerifiedHloModule>>
140 auto module = std::make_unique<VerifiedHloModule>( in ParseAndReturnVerifiedModule()
213 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> module, in RunAndFilecheckHloRewrite()
|
| D | buffer_donation_test.cc | 292 StatusOr<std::unique_ptr<VerifiedHloModule>> module = in TEST_F()
320 StatusOr<std::unique_ptr<VerifiedHloModule>> module = in TEST_F()
|
| D | llvm_compiler_test.cc | 169 return std::make_unique<VerifiedHloModule>( in CreateNewVerifiedModule()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | gpu_codegen_test.cc | 32 std::unique_ptr<VerifiedHloModule>
42 return std::make_unique<VerifiedHloModule>( in CreateNewVerifiedModuleWithFTZ()
49 std::unique_ptr<VerifiedHloModule> hlo_module, absl::string_view pattern) { in CompileAndOptionallyVerifyPtx()
|
| D | parallel_reduction_test.cc | 62 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> hlo_module, in TEST_F()
103 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> hlo_module, in TEST_F()
116 std::unique_ptr<VerifiedHloModule> module = CreateNewVerifiedModule(); in TEST_F()
218 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> hlo_module, in TEST_F()
|
| 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 | reduction_vectorization_test.cc | 79 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> optimized_module, in TEST_F()
146 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> optimized_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_compilation_parallelism_test.cc | 55 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> optimized_module, in TEST_F()
|
| D | gpu_copy_alone_test.cc | 48 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> optimized_module, in TEST_F()
|
| D | gpu_kernel_tiling_test.cc | 436 std::unique_ptr<VerifiedHloModule> hlo_module = in TEST_F()
715 std::unique_ptr<VerifiedHloModule> hlo_module = in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_pass_pipeline_test.cc | 39 TF_ASSIGN_OR_RETURN(std::unique_ptr<VerifiedHloModule> module, in ParseModuleGroup()
147 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> module, in TEST_F()
170 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> module, in TEST_F()
197 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> module, in TEST_F()
233 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> module, in TEST_F()
302 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> module, in TEST_F()
354 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> module, in TEST_F()
|
| D | while_loop_simplifier_test.cc | 48 [[nodiscard]] std::unique_ptr<VerifiedHloModule> MakeModuleWithSimpleLoop(
54 [[nodiscard]] std::unique_ptr<VerifiedHloModule>
58 std::unique_ptr<VerifiedHloModule>
91 std::unique_ptr<VerifiedHloModule>
|
| D | hlo_parser.h | 95 friend class VerifiedHloModule; variable
|
| D | hlo_creation_utils_test.cc | 32 std::unique_ptr<VerifiedHloModule> CreateModuleWithProgramShape( in CreateModuleWithProgramShape()
47 std::unique_ptr<VerifiedHloModule> CreateModuleWithProgramShape( in CreateModuleWithProgramShape()
|
| D | layout_assignment_test.cc | 1151 std::unique_ptr<VerifiedHloModule> m, in TEST_F()
1167 std::unique_ptr<VerifiedHloModule> m, in TEST_F()
1194 std::unique_ptr<VerifiedHloModule> m, in TEST_F()
1227 std::unique_ptr<VerifiedHloModule> m, in TEST_F()
1254 std::unique_ptr<VerifiedHloModule> m, in TEST_F()
1281 std::unique_ptr<VerifiedHloModule> m, in TEST_F()
1316 std::unique_ptr<VerifiedHloModule> m, in TEST_F()
|
| D | while_util_test.cc | 34 StatusOr<std::unique_ptr<VerifiedHloModule>> GetParsedModule( in GetParsedModule()
|
| D | conditional_simplifier_test.cc | 508 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<VerifiedHloModule> module, in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/ |
| D | conv_emitter_test.cc | 46 VerifiedHloModule hlo_module( in CompileHloConvAndGetMlir()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_support_utils_test.cc | 50 xla::VerifiedHloModule* module, absl::string_view target) { in GetCustomCall()
|