| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | compiler.h | 182 virtual StatusOr<std::unique_ptr<HloModule>> RunHloPasses(
185 StatusOr<std::unique_ptr<HloModule>> RunHloPasses( in RunHloPasses() function
188 return RunHloPasses(std::move(module), executor, in RunHloPasses()
|
| D | llvm_compiler.h | 71 using Compiler::RunHloPasses;
|
| D | llvm_compiler.cc | 46 RunHloPasses(std::move(modules[i]), stream_execs[i][0], in Compile()
|
| D | layout_assignment_test.cc | 681 ->RunHloPasses(m->Clone(), backend().default_stream_executor(), in TEST_F()
937 ->RunHloPasses(m->Clone(), backend().default_stream_executor(), in TEST_F()
969 ->RunHloPasses(m->Clone(), backend().default_stream_executor(), in TEST_F()
1002 ->RunHloPasses(m->Clone(), backend().default_stream_executor(), in TEST_F()
1035 ->RunHloPasses(m->Clone(), backend().default_stream_executor(), in TEST_F()
1059 ->RunHloPasses(m->Clone(), backend().default_stream_executor(), in TEST_F()
|
| D | service.cc | 771 TF_ASSIGN_OR_RETURN(module, backend->compiler()->RunHloPasses( in BuildExecutable()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.h | 133 StatusOr<std::unique_ptr<HloModule>> RunHloPasses(
161 Status RunHloPasses(HloModule* module, bool is_aot_compile,
|
| D | cpu_compiler.cc | 455 Status CpuCompiler::RunHloPasses(HloModule* module, bool is_aot_compile, in RunHloPasses() function in xla::cpu::CpuCompiler
574 StatusOr<std::unique_ptr<HloModule>> CpuCompiler::RunHloPasses( in RunHloPasses() function in xla::cpu::CpuCompiler
582 TF_RETURN_IF_ERROR(RunHloPasses(module.get(), /*is_aot_compile=*/false, in RunHloPasses()
595 RunHloPasses(std::move(module), executor, options)); in RunHloPassesAndBufferAssignement()
912 RunHloPasses(module, /*is_aot_compile=*/true, target_machine.get())); in CompileAheadOfTime()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | compiler.cc | 96 StatusOr<std::unique_ptr<HloModule>> InterpreterCompiler::RunHloPasses( in RunHloPasses() function in xla::interpreter::InterpreterCompiler
144 TF_ASSIGN_OR_RETURN(auto module, RunHloPasses(std::move(hlo_modules[0]), in Compile()
|
| D | compiler.h | 46 StatusOr<std::unique_ptr<HloModule>> RunHloPasses(
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_fused_conv_rewriter_test.cc | 38 ->RunHloPasses( in GetOptimizedHlo()
334 ->RunHloPasses( in TEST_F()
|
| D | gpu_compiler.h | 56 StatusOr<std::unique_ptr<HloModule>> RunHloPasses(
|
| D | gpu_compiler.cc | 489 StatusOr<std::unique_ptr<HloModule>> GpuCompiler::RunHloPasses( in RunHloPasses() function in xla::gpu::GpuCompiler
518 RunHloPasses(std::move(hlo_module), executor, options)); in RunHloPassesAndBufferAssignement()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | codegen_test_base.cc | 22 TF_ASSIGN_OR_RETURN(hlo_module, backend().compiler()->RunHloPasses( in CompileToExecutable()
|
| D | llvm_irgen_test_base.cc | 104 return backend().compiler()->RunHloPasses( in GetOptimizedModule()
|
| D | buffer_donation_test.cc | 69 TF_ASSERT_OK_AND_ASSIGN(hlo_module, backend_->compiler()->RunHloPasses( in RunAndCheck()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | gpu_convolution_regression_test.cc | 29 (void)backend().compiler()->RunHloPasses( in CheckForHloText()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_on_demand_compiler.cc | 218 StatusOr<std::unique_ptr<HloModule>> RunHloPasses( in RunHloPasses() function in xla::__anoncef202870111::TpuCompiler
|