Home
last modified time | relevance | path

Searched refs:run_hlo_passes (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dhlo_test_base.cc207 int64 num_replicas, bool use_threads, bool run_hlo_passes) { in ExecuteReplicated() argument
210 options.run_hlo_passes = run_hlo_passes; in ExecuteReplicated()
221 bool run_hlo_passes, bool use_threads) { in ExecuteReplicated() argument
224 options.run_hlo_passes = run_hlo_passes; in ExecuteReplicated()
237 int64 num_replicas, bool run_hlo_passes) { in ExecuteReplicated() argument
240 options.run_hlo_passes = run_hlo_passes; in ExecuteReplicated()
267 const optional<ErrorSpec>& error, bool run_hlo_passes, in RunAndCompareInternal() argument
276 test_runner_.Execute(std::move(module), arguments, run_hlo_passes)); in RunAndCompareInternal()
279 arguments, run_hlo_passes)); in RunAndCompareInternal()
341 bool run_hlo_passes) { in Run() argument
[all …]
Dhlo_test_base.h164 int64 num_replicas, bool use_threads, bool run_hlo_passes = false);
170 bool run_hlo_passes, bool use_threads);
177 int64 num_replicas, bool run_hlo_passes);
222 bool run_hlo_passes) TF_MUST_USE_RESULT;
233 bool run_hlo_passes = true,
239 const absl::string_view hlo_string, bool run_hlo_passes = true,
245 const absl::string_view hlo_string, bool run_hlo_passes,
337 const absl::optional<ErrorSpec>& error, bool run_hlo_passes,
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_runner_interface.h77 bool run_hlo_passes = false; member
114 std::unique_ptr<HloModule> module, bool run_hlo_passes) = 0;
123 bool run_hlo_passes = true) {
124 return Execute(std::move(module), arguments, run_hlo_passes, nullptr);
129 bool run_hlo_passes = true,
134 bool run_hlo_passes,
Dhlo_runner.h76 bool run_hlo_passes,
91 bool run_hlo_passes = true, ExecutionProfile* profile = nullptr);
100 std::unique_ptr<HloModule> module, bool run_hlo_passes) override;
Dhlo_runner.cc93 bool run_hlo_passes, in Execute() argument
101 /*run_hlo_passes=*/run_hlo_passes, in Execute()
152 absl::Span<ScopedShapedBuffer const> arguments, bool run_hlo_passes, in ExecuteWithDeviceBuffers() argument
155 CreateExecutable(std::move(module), run_hlo_passes)); in ExecuteWithDeviceBuffers()
188 CreateExecutable(std::move(module), options.run_hlo_passes)); in ExecuteReplicated()
427 std::unique_ptr<HloModule> module, bool run_hlo_passes) { in CreateExecutable() argument
428 if (run_hlo_passes) { in CreateExecutable()
Dhlo_runner_interface.cc90 bool run_hlo_passes, ExecutionProfile* profile) { in Execute() argument
100 /*run_hlo_passes=*/run_hlo_passes, in Execute()
/external/tensorflow/tensorflow/compiler/xla/tools/
Drun_hlo_module.cc92 se::Platform* platform, bool run_hlo_passes) { in ExecuteOnPlatform() argument
102 auto result_status = runner.Execute(std::move(module), args, run_hlo_passes); in ExecuteOnPlatform()