Searched defs:module_group (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | llvm_compiler.cc | 25 HloModuleGroup* module_group, in RunHloPassesOnModuleGroup()
34 std::unique_ptr<HloModuleGroup> module_group, in RunBackendOnModuleGroup()
42 std::unique_ptr<HloModuleGroup> module_group, in Compile()
|
| D | hlo_pass_pipeline.cc | 107 void HloPassPipeline::MaybeDumpHlo(const HloModuleGroup& module_group, in MaybeDumpHlo()
125 StatusOr<bool> HloPassPipeline::RunOnModuleGroup(HloModuleGroup* module_group) { in RunOnModuleGroup()
|
| D | hlo_pass_interface.h | 51 StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group) override { in RunOnModuleGroup()
|
| D | hlo_pass_fix.h | 59 StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group) override { in RunOnModuleGroup()
|
| D | hlo_pass_pipeline.h | 100 HloModuleGroup* module_group) { in RunHelper()
|
| D | compiler.cc | 48 std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime()
|
| D | hlo_pass_pipeline_test.cc | 65 StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group) override { in RunOnModuleGroup()
|
| D | hlo_runner.cc | 429 auto module_group = absl::make_unique<HloModuleGroup>(std::move(module)); in CreateExecutable() local
|
| D | service.cc | 367 auto module_group = in BuildExecutables() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | compiler.cc | 106 HloModuleGroup* module_group, in RunHloPassesOnModuleGroup()
138 std::unique_ptr<HloModuleGroup> module_group, in RunBackendOnModuleGroup()
146 std::unique_ptr<HloModuleGroup> module_group, in Compile()
174 std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | codegen_test_base.cc | 35 auto module_group = absl::make_unique<HloModuleGroup>(std::move(hlo_module)); in CompileToAotCompilationResult() local
|
| D | llvm_compiler_test.cc | 96 auto module_group = absl::make_unique<HloModuleGroup>("test_module_group"); in TestMultiModuleCompilation() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nvptx_compiler.cc | 880 NVPTXCompiler::CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.cc | 729 CpuCompiler::CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime()
|