Searched defs:module_group (Results 1 – 17 of 17) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_pass_pipeline.cc | 44 void RecordPassStartMetadata(HloModuleGroup& module_group, in RecordPassStartMetadata()
74 Status AttemptRecordPassEndMetadata(HloModuleGroup& module_group, in AttemptRecordPassEndMetadata()
89 void RecordPassEndMetadata(HloModuleGroup& module_group, in RecordPassEndMetadata()
115 void SetInstructionMetadata(HloModuleGroup& module_group) { in SetInstructionMetadata()
250 HloModuleGroup& module_group, absl::string_view after_pass_name, in MaybeDumpHloAndSaveFilenames()
267 StatusOr<bool> HloPassPipeline::RunOnModuleGroup(HloModuleGroup* module_group) { in RunOnModuleGroup()
|
| D | hlo_pass_interface.h | 53 StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group) override { in RunOnModuleGroup()
|
| D | llvm_compiler.cc | 26 std::unique_ptr<HloModuleGroup> module_group, in Compile()
|
| D | hlo_pass_fix.h | 60 StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group) override { in RunOnModuleGroup()
|
| D | hlo_pass_pipeline_test.cc | 69 StatusOr<bool> RunOnModuleGroup(HloModuleGroup* module_group) override { in RunOnModuleGroup()
264 HloModuleGroup module_group(TestName()); in TEST_F() local
|
| D | hlo_pass_pipeline.h | 122 HloModuleGroup* module_group) { in RunHelper()
|
| D | compiler.cc | 48 std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime()
|
| D | compiler.h | 232 std::unique_ptr<HloModuleGroup> module_group, in Compile()
|
| D | hlo_runner.cc | 429 auto module_group = absl::make_unique<HloModuleGroup>(std::move(module)); in CreateExecutable() local
|
| D | service.cc | 306 auto module_group = in BuildExecutables() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | compiler.cc | 129 std::unique_ptr<HloModuleGroup> module_group, in Compile()
155 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 | 138 auto module_group = absl::make_unique<HloModuleGroup>("test_module_group"); in TestMultiModuleCompilation() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | vectorized_reduce_with_no_vector_registers_test.cc | 74 auto module_group = absl::make_unique<HloModuleGroup>("group"); in TEST_F() local
|
| D | cpu_compiler.cc | 191 std::unique_ptr<HloModuleGroup> module_group, in Compile()
804 CpuCompiler::CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_on_demand_compiler.cc | 276 std::unique_ptr<HloModuleGroup> module_group, in Compile()
351 CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_compiler.cc | 978 GpuCompiler::CompileAheadOfTime(std::unique_ptr<HloModuleGroup> module_group, in CompileAheadOfTime()
|