Home
last modified time | relevance | path

Searched refs:debug_options (Results 1 – 25 of 81) sorted by relevance

1234

/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_unrolling_test.cc48 auto debug_options = HloTestBase::GetDebugOptionsForTest(); in TEST_F() local
49 debug_options.set_xla_gpu_max_kernel_unroll_factor(1); in TEST_F()
50 config.set_debug_options(debug_options); in TEST_F()
65 auto debug_options = HloTestBase::GetDebugOptionsForTest(); in TEST_F() local
68 debug_options.set_xla_gpu_max_kernel_unroll_factor(8); in TEST_F()
69 config.set_debug_options(debug_options); in TEST_F()
108 auto debug_options = HloTestBase::GetDebugOptionsForTest(); in TEST_F() local
109 debug_options.set_xla_gpu_max_kernel_unroll_factor(4); in TEST_F()
110 config.set_debug_options(debug_options); in TEST_F()
139 auto debug_options = HloTestBase::GetDebugOptionsForTest(); in TEST_F() local
[all …]
Dgpu_codegen_test.cc29 auto debug_options = GetDebugOptionsFromFlags(); in CreateNewUnverifiedModuleWithFTZ() local
30 debug_options.set_xla_gpu_ftz(ftz); in CreateNewUnverifiedModuleWithFTZ()
31 debug_options.set_xla_gpu_max_kernel_unroll_factor(1); in CreateNewUnverifiedModuleWithFTZ()
33 debug_options.add_xla_disable_hlo_passes("constant_folding"); in CreateNewUnverifiedModuleWithFTZ()
34 config.set_debug_options(debug_options); in CreateNewUnverifiedModuleWithFTZ()
Dgpu_index_test.cc120 auto debug_options = HloTestBase::GetDebugOptionsForTest(); in TEST_F() local
121 debug_options.set_xla_gpu_max_kernel_unroll_factor(1); in TEST_F()
122 config.set_debug_options(debug_options); in TEST_F()
Dgpu_kernel_tiling_test.cc40 auto debug_options = HloTestBase::GetDebugOptionsForTest(); in ConfigWithoutLayoutAssignment() local
42 debug_options.add_xla_disable_hlo_passes("layout-assignment"); in ConfigWithoutLayoutAssignment()
43 config.set_debug_options(debug_options); in ConfigWithoutLayoutAssignment()
/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_index_splitter_test.cc44 DebugOptions debug_options = config.debug_options(); in TEST_F() local
45 debug_options.set_xla_allow_scalar_index_dynamic_ops(true); in TEST_F()
46 config.set_debug_options(debug_options); in TEST_F()
81 DebugOptions debug_options = config.debug_options(); in TEST_F() local
82 debug_options.set_xla_allow_scalar_index_dynamic_ops(true); in TEST_F()
83 config.set_debug_options(debug_options); in TEST_F()
120 DebugOptions debug_options = config.debug_options(); in TEST_F() local
121 debug_options.set_xla_allow_scalar_index_dynamic_ops(true); in TEST_F()
122 config.set_debug_options(debug_options); in TEST_F()
Dhlo_pass_pipeline.cc77 const DebugOptions& debug_options) { in GetEnabledPasses() argument
78 auto repeated_field = debug_options.xla_disable_hlo_passes(); in GetEnabledPasses()
81 if (debug_options.xla_disable_all_hlo_passes()) { in GetEnabledPasses()
122 GetEnabledPasses(module->config().debug_options())); in Run()
138 GetEnabledPasses(module_group->module(0).config().debug_options())); in RunOnModuleGroup()
Ddump.cc208 /*label=*/filename, module.config().debug_options(), format, profile); in DumpHloModuleImpl()
255 CanonicalDebugOptions(module.config().debug_options())); in DumpToFileInDir()
262 CanonicalDebugOptions(module.config().debug_options())); in DumpToFileInDirOrStdout()
266 CanonicalDebugOptions opts(module.config().debug_options()); in DumpHloModuleIfEnabled()
275 CanonicalDebugOptions opts(module.config().debug_options()); in DumpHloModuleIfEnabled()
284 CanonicalDebugOptions opts(module.config().debug_options()); in DumpHloModuleIfEnabled()
303 CanonicalDebugOptions opts(module.config().debug_options()); in DumpHloModuleBetweenPassesIfEnabled()
329 CanonicalDebugOptions opts(module.config().debug_options()); in DumpHloModuleDuringPassIfEnabled()
349 CanonicalDebugOptions opts(module.config().debug_options()); in DumpHloSnapshotIfEnabled()
Delemental_ir_emitter_test.cc87 auto debug_options = GetDebugOptionsForTest(); in XLA_TEST_F() local
90 debug_options.add_xla_disable_hlo_passes("layout-assignment"); in XLA_TEST_F()
91 config.set_debug_options(debug_options); in XLA_TEST_F()
Dhlo_module_config.h90 const DebugOptions& debug_options() const { return debug_options_; } in debug_options() function
92 void set_debug_options(const DebugOptions& debug_options) { in set_debug_options() argument
93 debug_options_ = debug_options; in set_debug_options()
Dhlo_runner.cc38 const DebugOptions& debug_options) { in CreateModuleFromString() argument
40 config.set_debug_options(debug_options); in CreateModuleFromString()
48 const HloProto& proto, const DebugOptions& debug_options) { in HloProtoToModule() argument
51 debug_options)); in HloProtoToModule()
61 const DebugOptions& debug_options) { in ReadModuleFromBinaryProtoFile() argument
65 return HloProtoToModule(proto, debug_options); in ReadModuleFromBinaryProtoFile()
70 const DebugOptions& debug_options) { in ReadModuleFromTextProtoFile() argument
74 return HloProtoToModule(proto, debug_options); in ReadModuleFromTextProtoFile()
79 const DebugOptions& debug_options) { in ReadModuleFromHloTextFile() argument
84 config.set_debug_options(debug_options); in ReadModuleFromHloTextFile()
Dhlo_runner.h90 const absl::string_view hlo_string, const DebugOptions& debug_options);
95 const std::string& filename, const DebugOptions& debug_options);
97 const std::string& filename, const DebugOptions& debug_options);
102 const std::string& filename, const DebugOptions& debug_options);
Dhlo_verifier_test.cc404 DebugOptions debug_options = config.debug_options(); in TEST_F() local
405 debug_options.set_xla_allow_scalar_index_dynamic_ops(true); in TEST_F()
406 config.set_debug_options(debug_options); in TEST_F()
430 DebugOptions debug_options = config.debug_options(); in TEST_F() local
431 debug_options.set_xla_allow_scalar_index_dynamic_ops(true); in TEST_F()
432 config.set_debug_options(debug_options); in TEST_F()
Dcompile_only_service.cc73 const DebugOptions& debug_options = options.debug_options(); in CompileAheadOfTime() local
75 *execution_options.mutable_debug_options() = debug_options; in CompileAheadOfTime()
/external/tensorflow/tensorflow/python/debug/lib/
Ddebug_utils_test.py103 debug_watch_opts = self._run_options.debug_options.debug_tensor_watch_opts
129 debug_watch_opts = self._run_options.debug_options.debug_tensor_watch_opts
151 debug_watch_opts = self._run_options.debug_options.debug_tensor_watch_opts
173 debug_watch_opts = self._run_options.debug_options.debug_tensor_watch_opts
196 debug_watch_opts = self._run_options.debug_options.debug_tensor_watch_opts
229 self._run_options.debug_options.debug_tensor_watch_opts, 0,
244 self._run_options.debug_options.debug_tensor_watch_opts, 0,
257 self._run_options.debug_options.debug_tensor_watch_opts, 0,
270 self._run_options.debug_options.debug_tensor_watch_opts, 0,
284 self._run_options.debug_options.debug_tensor_watch_opts, 0,
[all …]
Ddebug_utils.py61 watch_opts = run_options.debug_options.debug_tensor_watch_opts
62 run_options.debug_options.global_step = global_step
176 run_options.debug_options.reset_disk_byte_usage = reset_disk_byte_usage
269 run_options.debug_options.reset_disk_byte_usage = reset_disk_byte_usage
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_options.cc38 config.debug_options().xla_backend_extra_options(); in OptimizeForSizeRequested()
44 config.debug_options().xla_backend_extra_options(); in VectorizedReduceDisabled()
50 config.debug_options().xla_backend_extra_options(); in LlvmIrGemvTilingFactor()
62 config.debug_options().xla_backend_extra_options(); in ForceEnableExperimentalLlvmIrGemm()
76 config.debug_options().xla_backend_extra_options(); in LlvmIrGemmTileSize()
Dcpu_compiler.cc257 &pipeline, module->config().debug_options(), in RunHloPassesThroughLayoutAssn()
332 &pipeline, module->config().debug_options(), in RunHloPassesThroughLayoutAssn()
413 if (module_config.debug_options().xla_cpu_enable_fast_math()) { in CompilerTargetOptions()
416 module_config.debug_options().xla_cpu_fast_math_honor_infs(); in CompilerTargetOptions()
418 module_config.debug_options().xla_cpu_fast_math_honor_nans(); in CompilerTargetOptions()
431 << module_config.debug_options().xla_backend_optimization_level(); in CodeGenOptLevel()
432 switch (module_config.debug_options().xla_backend_optimization_level()) { in CodeGenOptLevel()
610 module->config().debug_options().xla_cpu_enable_fast_math(), in RunBackend()
611 module->config().debug_options().xla_llvm_disable_expensive_passes(), in RunBackend()
633 module->config().debug_options().xla_embed_ir_in_executable(); in RunBackend()
[all …]
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtoken_hlo_test.cc144 DebugOptions debug_options = GetDebugOptionsForTest(); in XLA_TEST_F() local
146 debug_options.add_xla_disable_hlo_passes("hlo-module-dce"); in XLA_TEST_F()
149 HloRunner::CreateModuleFromString(module_string, debug_options)); in XLA_TEST_F()
179 DebugOptions debug_options = GetDebugOptionsForTest(); in XLA_TEST_F() local
181 debug_options.add_xla_disable_hlo_passes("hlo-module-dce"); in XLA_TEST_F()
187 HloRunner::CreateModuleFromString(module_string, debug_options)); in XLA_TEST_F()
197 HloRunner::CreateModuleFromString(module_string, debug_options)); in XLA_TEST_F()
Dhlo_test_base.cc182 auto debug_options = GetDebugOptionsFromFlags(); in GetDebugOptionsForTest() local
184 debug_options.add_xla_disable_hlo_passes("constant_folding"); in GetDebugOptionsForTest()
185 debug_options.set_xla_gpu_max_kernel_unroll_factor(1); in GetDebugOptionsForTest()
186 debug_options.set_xla_hlo_evaluator_use_fast_path(true); in GetDebugOptionsForTest()
187 return debug_options; in GetDebugOptionsForTest()
369 DebugOptions debug_options = config.debug_options(); in Run() local
370 debug_options.set_xla_hlo_profile(true); in Run()
371 config.set_debug_options(debug_options); in Run()
423 DebugOptions debug_options = config.debug_options(); in RunMultipleTimes() local
424 debug_options.set_xla_hlo_profile(true); in RunMultipleTimes()
[all …]
/external/tensorflow/tensorflow/core/debug/
Ddebugger_state_impl.h28 DebuggerState(const DebugOptions& debug_options);
48 DebugGraphDecorator(const DebugOptions& debug_options) in DebugGraphDecorator() argument
49 : debug_options_(debug_options) {} in DebugGraphDecorator()
Ddebugger_state_impl.cc23 DebuggerState::DebuggerState(const DebugOptions& debug_options) { in DebuggerState() argument
25 debug_options.debug_tensor_watch_opts()) { in DebuggerState()
30 if (debug_options.reset_disk_byte_usage()) { in DebuggerState()
Dgrpc_session_debug_test.cc169 DebugOptions* debug_options = options.mutable_debug_options(); in TEST_F() local
170 DebugTensorWatch* watch = debug_options->add_debug_tensor_watch_opts(); in TEST_F()
177 watch = debug_options->add_debug_tensor_watch_opts(); in TEST_F()
255 DebugOptions* debug_options = options.mutable_debug_options(); in TEST_F() local
256 DebugTensorWatch* watch = debug_options->add_debug_tensor_watch_opts(); in TEST_F()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dgraph_mgr.h79 const DebugOptions& debug_options, int64 collective_graph_key,
182 const DebugOptions& debug_options, int64 collective_graph_key,
185 Status DecorateAndPublishGraphForDebug(const DebugOptions& debug_options,
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dalias_analysis.cc54 if (module_.config().debug_options().xla_llvm_enable_alias_scope_metadata()) { in AddAliasingInformationToIrArray()
65 if (module_.config().debug_options().xla_llvm_enable_noalias_metadata()) { in AddAliasingInformationToIrArray()
77 .debug_options() in AddAliasingInformationToIrArray()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dstream_assignment_test.cc35 auto debug_options = GetDebugOptionsForTest(); in CreateNewVerifiedModule() local
36 debug_options.set_xla_gpu_disable_multi_streaming(false); in CreateNewVerifiedModule()
37 config.set_debug_options(debug_options); in CreateNewVerifiedModule()

1234