/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | hlo_test_base.h | 82 std::unique_ptr<HloModule> CreateNewUnverifiedModule( 100 HloModule* module); 146 StatusOr<Literal> Execute(std::unique_ptr<HloModule> module, 151 Literal ExecuteNoHloPasses(std::unique_ptr<HloModule> module, 154 Literal ExecuteAndTransfer(std::unique_ptr<HloModule> module, 163 std::unique_ptr<HloModule> module, absl::Span<Literal* const> arguments, 168 std::unique_ptr<HloModule> module, absl::Span<Literal* const> arguments, 191 std::unique_ptr<HloModule> module, 194 const std::function<void(HloModule*)>& reference_preprocessor = nullptr) 200 std::unique_ptr<HloModule> module, [all …]
|
D | hlo_test_base.cc | 75 ProgramShape GetProgramShapeWithLayout(const HloModule& module) { in GetProgramShapeWithLayout() 114 std::unique_ptr<HloModule> HloTestBase::CreateNewUnverifiedModule( in CreateNewUnverifiedModule() 116 return absl::make_unique<HloModule>(name, GetModuleConfigForTest()); in CreateNewUnverifiedModule() 147 HloModule* module) { in RunHloPass() 187 StatusOr<Literal> HloTestBase::Execute(std::unique_ptr<HloModule> module, in Execute() 192 Literal HloTestBase::ExecuteNoHloPasses(std::unique_ptr<HloModule> module, in ExecuteNoHloPasses() 200 Literal HloTestBase::ExecuteAndTransfer(std::unique_ptr<HloModule> module, in ExecuteAndTransfer() 206 std::unique_ptr<HloModule> module, absl::Span<Literal* const> arguments, in ExecuteReplicated() 219 std::unique_ptr<HloModule> module, absl::Span<Literal* const> arguments, in ExecuteReplicated() 246 StatusOr<std::unique_ptr<HloModule>> HloTestBase::MakeReferenceModule( in MakeReferenceModule() [all …]
|
D | token_hlo_test.cc | 31 std::unique_ptr<HloModule> module = CreateNewVerifiedModule(); in XLA_TEST_F() 42 std::unique_ptr<HloModule> module = CreateNewVerifiedModule(); in XLA_TEST_F() 56 std::unique_ptr<HloModule> module = CreateNewVerifiedModule(); in XLA_TEST_F() 71 std::unique_ptr<HloModule> module = CreateNewUnverifiedModule(); in XLA_TEST_F() 92 std::unique_ptr<HloModule> module = CreateNewUnverifiedModule(); in XLA_TEST_F() 147 std::unique_ptr<HloModule> module, in XLA_TEST_F() 185 std::unique_ptr<HloModule> module, in XLA_TEST_F() 195 std::unique_ptr<HloModule> module, in XLA_TEST_F() 223 std::unique_ptr<HloModule> module, in XLA_TEST_F() 244 std::unique_ptr<HloModule> module, in XLA_TEST_F() [all …]
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_parser_test.cc | 103 R"(HloModule module in CreateTestCases() 115 R"(HloModule constant_s32_module in CreateTestCases() 127 R"(HloModule ConstantF32_module in CreateTestCases() 138 R"(HloModule ConstantF32Empty_module in CreateTestCases() 149 R"(HloModule ConstantF32R4Empty_module in CreateTestCases() 160 R"(HloModule Small_3x2x1x1_module in CreateTestCases() 171 R"(HloModule IsFiniteR1F32s_module in CreateTestCases() 183 R"(HloModule ConstantF16_module in CreateTestCases() 194 R"(HloModule BF16 in CreateTestCases() 205 R"(HloModule add_constants_module in CreateTestCases() [all …]
|
D | dump.h | 38 string TimestampFor(const HloModule& module); 41 string FilenameFor(const HloModule& module, absl::string_view prefix, 48 void DumpToFileInDir(const HloModule& module, absl::string_view file_prefix, 53 void DumpToFileInDirOrStdout(const HloModule& module, 69 void DumpHloModuleIfEnabled(const HloModule& module, absl::string_view name); 70 void DumpHloModuleIfEnabled(const HloModule& module, 73 void DumpHloModuleIfEnabled(const HloModule& module, 82 absl::string_view after_pass_name, const HloModule& module); 90 const HloModule& module); 98 void DumpHloSnapshotIfEnabled(const HloModule& module, [all …]
|
D | hlo_module_group.h | 38 explicit HloModuleGroup(std::unique_ptr<HloModule> module); 42 absl::Span<std::unique_ptr<HloModule>> modules); 44 std::vector<std::unique_ptr<HloModule>>&& modules); 47 const std::vector<HloModule*>& modules() const { return module_ptrs_; } in modules() 50 HloModule& module(int index) const { return *module_ptrs_.at(index); } in module() 53 void push_back(std::unique_ptr<HloModule> module); 57 void ReplaceModule(int index, std::unique_ptr<HloModule> module); 61 std::vector<std::unique_ptr<HloModule>> ConsumeModules(); 92 std::vector<std::unique_ptr<HloModule>> modules_; 96 std::vector<HloModule*> module_ptrs_;
|
D | hlo_module_group.cc | 22 HloModuleGroup::HloModuleGroup(std::unique_ptr<HloModule> module) in HloModuleGroup() 28 absl::Span<std::unique_ptr<HloModule>> modules) in HloModuleGroup() 36 absl::string_view name, std::vector<std::unique_ptr<HloModule>>&& modules) in HloModuleGroup() 43 std::vector<std::unique_ptr<HloModule>> HloModuleGroup::ConsumeModules() { in ConsumeModules() 44 std::vector<std::unique_ptr<HloModule>> ret_modules = std::move(modules_); in ConsumeModules() 55 for (const HloModule* module : modules()) { in ToString() 64 for (const HloModule* module : modules()) { in ToProto() 86 std::vector<std::unique_ptr<HloModule>> modules; in CreateFromProto() 90 std::unique_ptr<HloModule> module, in CreateFromProto() 91 HloModule::CreateFromProto(module_proto, module_configs[i])); in CreateFromProto() [all …]
|
D | hlo_runner_interface.cc | 22 /*static*/ StatusOr<std::unique_ptr<HloModule>> 33 StatusOr<std::unique_ptr<HloModule>> HloProtoToModule( in HloProtoToModule() 36 HloModule::CreateModuleConfigFromProto(proto.hlo_module(), in HloProtoToModule() 39 HloModule::CreateFromProto(proto.hlo_module(), config)); in HloProtoToModule() 45 /*static*/ StatusOr<std::unique_ptr<HloModule>> 54 /*static*/ StatusOr<std::unique_ptr<HloModule>> 63 /*static*/ StatusOr<std::unique_ptr<HloModule>> 74 /*static*/ StatusOr<std::unique_ptr<HloModule>> 83 HloModule::CreateModuleConfigFromProto(module_proto, debug_options)); in ReadModuleFromModuleBinaryProtofile() 85 return HloModule::CreateFromProto(module_proto, module_config); in ReadModuleFromModuleBinaryProtofile() [all …]
|
D | hlo_runner_interface.h | 90 static StatusOr<std::unique_ptr<HloModule>> CreateModuleFromString( 95 static StatusOr<std::unique_ptr<HloModule>> ReadModuleFromBinaryProtoFile( 97 static StatusOr<std::unique_ptr<HloModule>> ReadModuleFromTextProtoFile( 102 static StatusOr<std::unique_ptr<HloModule>> 108 static StatusOr<std::unique_ptr<HloModule>> ReadModuleFromHloTextFile( 114 std::unique_ptr<HloModule> module, bool run_hlo_passes) = 0; 121 StatusOr<Literal> Execute(std::unique_ptr<HloModule> module, 127 StatusOr<Literal> Execute(std::unique_ptr<HloModule> module, 132 virtual StatusOr<Literal> Execute(std::unique_ptr<HloModule> module, 158 std::unique_ptr<HloModule> module, [all …]
|
D | hlo_module.cc | 44 HloModule::HloModule(const string& name, HloModuleConfig config) in HloModule() function in xla::HloModule 52 Status HloModule::set_schedule(HloSchedule schedule) { in set_schedule() 59 void HloModule::ReplaceEntryComputation(HloComputation* entry_computation) { in ReplaceEntryComputation() 67 HloComputation* HloModule::AddComputationInternal( in AddComputationInternal() 122 HloComputation* HloModule::AddEntryComputation( in AddEntryComputation() 129 HloComputation* HloModule::AddEntryComputationWithLayouts( in AddEntryComputationWithLayouts() 136 Status HloModule::RemoveEmbeddedComputation(HloComputation* to_remove) { in RemoveEmbeddedComputation() 151 HloComputation* HloModule::AddEmbeddedComputation( in AddEmbeddedComputation() 158 void HloModule::ReplaceComputations( in ReplaceComputations() 235 string HloModule::ToString(const HloPrintOptions& options) const { in ToString() [all …]
|
D | ar_crs_combiner_test.cc | 41 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F() 61 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F() 80 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F() 101 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F() 121 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F() 141 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F() 163 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F() 185 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F() 208 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F() 244 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F() [all …]
|
D | copy_insertion.h | 61 StatusOr<bool> Run(HloModule* module) override; 67 HloModule* module); 79 Status AddSpecialCaseCopies(HloModule* module); 84 HloModule* module); 95 Status AddCopiesToResolveInterference(HloModule* module);
|
D | while_loop_analysis_test.cc | 52 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F() 83 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F() 116 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F() 151 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F() 190 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F() 229 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F() 264 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> module, in TEST_F()
|
D | hlo_pass_pipeline.cc | 36 void RecordPassStartMetadata(HloModule& module, const std::string& pass_name, in RecordPassStartMetadata() 47 for (HloModule* module : module_group.modules()) { in RecordPassStartMetadata() 52 Status AttemptRecordPassEndMetadata(HloModule& module, in AttemptRecordPassEndMetadata() 65 void RecordPassEndMetadata(HloModule& module, const std::string& pass_name, in RecordPassEndMetadata() 77 for (HloModule* module : module_group.modules()) { in AttemptRecordPassEndMetadata() 78 for (HloModule* other_module : module_group.modules()) { in AttemptRecordPassEndMetadata() 98 void SetInstructionMetadata(HloModule& module) { in SetInstructionMetadata() 116 for (HloModule* module : module_group.modules()) { in SetInstructionMetadata() 238 HloModule& module, absl::string_view after_pass_name, in MaybeDumpHloAndSaveFilenames() 252 for (HloModule* module : module_group.modules()) { in MaybeDumpHloAndSaveFilenames() [all …]
|
D | batch_dot_simplification_test.cc | 40 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F() 64 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F() 88 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F() 112 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F() 136 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F() 160 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F() 187 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F() 208 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
|
D | hlo_proto_util.cc | 25 HloProto MakeHloProto(const HloModule& module, in MakeHloProto() 33 HloProto MakeHloProto(const HloModule& module) { in MakeHloProto() 40 StatusOr<std::unique_ptr<HloModule>> CreateModuleFromProto( in CreateModuleFromProto() 44 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloModule> module, in CreateModuleFromProto() 45 HloModule::CreateFromProto(proto, module_config)); in CreateModuleFromProto()
|
D | hlo_memory_scheduler.h | 52 HloModule*, const TuplePointsToAnalysis&, const HloAliasAnalysis&, 106 HloModule* module, const TuplePointsToAnalysis& points_to_analysis, 115 HloModule* module, const LogicalBuffer::SizeFunction& size_function, 140 StatusOr<bool> Run(HloModule* module) override; 154 StatusOr<bool> Run(HloModule* module) override; 165 StatusOr<bool> Run(HloModule* module) override;
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | gpu_compiler.h | 56 StatusOr<std::unique_ptr<HloModule>> RunHloPasses( 57 std::unique_ptr<HloModule> module, se::StreamExecutor* stream_exec, 61 std::tuple<std::unique_ptr<HloModule>, std::unique_ptr<BufferAssignment>>> 62 RunHloPassesAndBufferAssignement(std::unique_ptr<HloModule> hlo_module, 66 Status OptimizeHloModule(HloModule* hlo_module, 71 HloModule* hlo_module, se::StreamExecutor* stream_exec, 75 HloModule* hlo_module, se::StreamExecutor* stream_exec, 92 const HloModule* debug_module) = 0; 94 Status PrepareHloModuleForIrEmitting(HloModule* hlo_module); 97 std::unique_ptr<HloModule> module, se::StreamExecutor* stream_exec, [all …]
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | parallel_task_assignment_test.cc | 42 StatusOr<bool> RunParallelTaskAssigner(HloModule* module) { in RunParallelTaskAssigner() 60 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F() 88 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F() 105 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F() 122 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F() 167 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F() 188 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F() 202 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<HloModule> m, in TEST_F()
|
D | cpu_compiler.h | 133 StatusOr<std::unique_ptr<HloModule>> RunHloPasses( 134 std::unique_ptr<HloModule> module, se::StreamExecutor* stream_exec, 138 std::tuple<std::unique_ptr<HloModule>, std::unique_ptr<BufferAssignment>>> 139 RunHloPassesAndBufferAssignement(std::unique_ptr<HloModule> module, 144 std::unique_ptr<HloModule> module, se::StreamExecutor* stream_exec, 161 Status RunHloPasses(HloModule* module, bool is_aot_compile, 166 HloModule* module, bool /*is_aot_compile*/, 171 HloModule* module, bool is_aot_compile,
|
D | parallel_task_assignment.h | 43 HloModule* module, 82 StatusOr<bool> Run(HloModule* module) override; 90 bool AssignParallelTasks(HloModule* module, 93 HloModule* module, HloComputation* computation, 98 void ComputeTargetParallelTasks(HloModule* module,
|
/external/tensorflow/tensorflow/compiler/xla/tools/ |
D | prepare_reference_module.cc | 33 StatusOr<std::unique_ptr<HloModule>> PrepareReferenceModule( in PrepareReferenceModule() 34 const HloModule& test_module, in PrepareReferenceModule() 37 const std::function<Status(const HloModule&, in PrepareReferenceModule() argument 39 HloModule*)>& module_modifier_hook) { in PrepareReferenceModule() 51 std::unique_ptr<HloModule> reference_module = in PrepareReferenceModule()
|
D | prepare_reference_module.h | 35 StatusOr<std::unique_ptr<HloModule>> PrepareReferenceModule( 36 const HloModule& test_module, 39 const std::function<Status(const HloModule&, 41 HloModule*)>& module_modifier_hook = {});
|
D | hlo_extractor.cc | 43 const HloModule& old_module, in ExtractionVisitor() 46 module_(absl::make_unique<HloModule>("extracted", config_)), in ExtractionVisitor() 97 HloModule* module() { return module_.get(); } in module() 99 std::unique_ptr<HloModule> ConsumeModule() { return std::move(module_); } in ConsumeModule() 102 const HloModule& old_module_; 104 std::unique_ptr<HloModule> module_; 137 std::unique_ptr<HloModule> ExtractModule(HloInstruction* instruction, in ExtractModule()
|
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/ |
D | export.mlir | 4 // CHECK: HloModule 17 // CHECK: HloModule 46 // CHECK: HloModule 74 // CHECK: HloModule 91 // CHECK: HloModule 106 // CHECK: HloModule 129 // CHECK: HloModule 141 // CHECK: HloModule 151 // CHECK: HloModule 165 // CHECK: HloModule [all …]
|