Searched refs:run_hlo_passes (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_runner.h | 80 bool run_hlo_passes = false; member
119 bool run_hlo_passes = true,
124 bool run_hlo_passes = true,
140 bool run_hlo_passes = true, ExecutionProfile* profile = nullptr);
145 bool run_hlo_passes = true, ExecutionProfile* profile = nullptr);
163 std::unique_ptr<HloModule> module, bool run_hlo_passes);
|
| D | hlo_runner.cc | 142 const absl::Span<const Literal* const> arguments, bool run_hlo_passes, in Execute() argument
150 /*run_hlo_passes=*/run_hlo_passes, in Execute()
157 bool run_hlo_passes, in Execute() argument
168 /*run_hlo_passes=*/run_hlo_passes, in Execute()
203 const absl::Span<const ShapedBuffer* const> arguments, bool run_hlo_passes, in ExecuteWithDeviceBuffers() argument
213 CreateExecutable(std::move(module), run_hlo_passes)); in ExecuteWithDeviceBuffers()
224 const absl::Span<const ScopedShapedBuffer> arguments, bool run_hlo_passes, in ExecuteWithDeviceBuffers() argument
234 /*run_hlo_passes=*/run_hlo_passes, in ExecuteWithDeviceBuffers()
277 CreateExecutable(std::move(module), options.run_hlo_passes)); in ExecuteReplicated()
427 std::unique_ptr<HloModule> module, bool run_hlo_passes) { in CreateExecutable() argument
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | hlo_test_base.cc | 223 bool run_hlo_passes, bool use_threads) { in ExecuteReplicated() argument
226 options.run_hlo_passes = run_hlo_passes; in ExecuteReplicated()
255 const optional<ErrorSpec>& error, bool run_hlo_passes, in RunAndCompareInternal() argument
264 test_runner_.Execute(std::move(module), arguments, run_hlo_passes)); in RunAndCompareInternal()
267 arguments, run_hlo_passes)); in RunAndCompareInternal()
343 bool run_hlo_passes, in Run() argument
384 /*run_hlo_passes=*/run_hlo_passes, in Run()
393 string_view hlo_string, bool run_hlo_passes, in RunMultipleTimes() argument
437 test_runner_.CreateExecutable(std::move(module), run_hlo_passes); in RunMultipleTimes()
|
| D | hlo_test_base.h | 189 bool run_hlo_passes, bool use_threads);
240 bool run_hlo_passes = true,
244 const absl::string_view hlo_string, bool run_hlo_passes,
334 const absl::optional<ErrorSpec>& error, bool run_hlo_passes,
|