Home
last modified time | relevance | path

Searched defs:debug_options (Results 1 – 25 of 72) sorted by relevance

123

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
H A Dgpu_unrolling_test.cc48 auto debug_options = HloTestBase::GetDebugOptionsForTest(); in TEST_F() local
66 auto debug_options = HloTestBase::GetDebugOptionsForTest(); in TEST_F() local
91 auto debug_options = GetDebugOptionsFromFlags(); in TEST_F() local
114 auto debug_options = HloTestBase::GetDebugOptionsForTest(); in TEST_F() local
147 auto debug_options = HloTestBase::GetDebugOptionsForTest(); in TEST_F() local
181 auto debug_options = HloTestBase::GetDebugOptionsForTest(); in TEST_F() local
215 auto debug_options = HloTestBase::GetDebugOptionsForTest(); in TEST_F() local
240 auto debug_options = HloTestBase::GetDebugOptionsForTest(); in TEST_F() local
265 auto debug_options = HloTestBase::GetDebugOptionsForTest(); in TEST_F() local
H A Dgemm_rewrite_test.cc132 DebugOptions debug_options = GetDebugOptionsForTest(); local
143 DebugOptions debug_options = GetDebugOptionsForTest(); local
484 DebugOptions debug_options = GetDebugOptionsForTest(); local
934 DebugOptions debug_options = GpuCodegenTest::GetDebugOptionsForTest(); local
H A Dgpu_index_test.cc158 auto debug_options = HloTestBase::GetDebugOptionsForTest(); in TEST_F() local
187 auto debug_options = HloTestBase::GetDebugOptionsForTest(); in TEST_F() local
H A Dgpu_compilation_parallelism_test.cc29 DebugOptions debug_options = GpuCodegenTest::GetDebugOptionsForTest(); in GetDebugOptionsForTest() local
H A Dgpu_copy_alone_test.cc32 DebugOptions debug_options = GpuCodegenTest::GetDebugOptionsForTest(); in GetDebugOptionsForTest() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_runner_interface.cc24 const DebugOptions& debug_options) { in CreateModuleFromString()
34 const HloProto& proto, const DebugOptions& debug_options) { in HloProtoToModule()
47 const std::string& filename, const DebugOptions& debug_options) { in ReadModuleFromBinaryProtoFile()
56 const std::string& filename, const DebugOptions& debug_options) { in ReadModuleFromTextProtoFile()
65 const std::string& filename, const DebugOptions& debug_options) { in ReadModuleFromHloTextFile()
76 const std::string& filename, const DebugOptions& debug_options) { in ReadModuleFromModuleBinaryProtofile()
H A Delemental_ir_emitter_test.cc41 auto debug_options = GetDebugOptionsForTest(); in RunTypeConversionTest() local
122 auto debug_options = GetDebugOptionsForTest(); in XLA_TEST_F() local
147 auto debug_options = GetDebugOptionsForTest(); in XLA_TEST_F() local
170 auto debug_options = GetDebugOptionsForTest(); in XLA_TEST_F() local
194 auto debug_options = GetDebugOptionsForTest(); in XLA_TEST_F() local
H A Ddynamic_index_splitter_test.cc44 DebugOptions debug_options = config.debug_options(); in TEST_F() local
82 DebugOptions debug_options = config.debug_options(); in TEST_F() local
121 DebugOptions debug_options = config.debug_options(); in TEST_F() local
H A Ddump.cc577 void DumpToFileInDir(const DebugOptions& debug_options, in DumpToFileInDir()
589 void DumpToFileInDirOrStdout(const DebugOptions& debug_options, int unique_id, in DumpToFileInDirOrStdout()
620 const DebugOptions& debug_options, in DumpProtobufToFile()
652 const DebugOptions& debug_options, in DumpPerModuleProtobufToFile()
H A Dhlo_pass_pipeline.cc150 HloT* hlo, const DebugOptions& debug_options, in RunPassesInternal()
207 const DebugOptions& debug_options) { in GetEnabledPasses()
H A Dhlo_module_config.h208 const DebugOptions& debug_options() const { return debug_options_; } in debug_options() function
210 void set_debug_options(const DebugOptions& debug_options) { in set_debug_options()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgemm_algorithm_picker.cc59 AutotuneConfig GetConfig(const DebugOptions& debug_options) { in GetConfig()
66 const DebugOptions& debug_options, const AutotuneConfig& config) { in CreateRedzoneAllocator()
249 const DebugOptions& debug_options = in DoGemmAutotune() local
H A Dgpu_executable.cc151 DebugOptions debug_options) { in Create()
176 const DebugOptions& debug_options() const { return debug_options_; } in debug_options() function in xla::gpu::GpuExecutable::JitRtExecutable
181 DebugOptions debug_options) in JitRtExecutable()
189 DebugOptions debug_options) in JitRtExecutable()
1028 xla::EntryFunctionAttributes entry_func_attrs, DebugOptions debug_options, in LoadFromObjFile()
H A Djitrt_custom_calls.cc515 const DebugOptions* debug_options, in operator ()()
591 const DebugOptions* debug_options, runtime::StridedMemrefView a, in operator ()()
1312 const DebugOptions* debug_options, in operator ()()
1391 const DebugOptions* debug_options, in run()
1419 const DebugOptions* debug_options, runtime::StridedMemrefView a, in operator ()()
1505 const DebugOptions* debug_options, in operator ()()
H A Dgpu_asm_opts_util.cc29 const DebugOptions& debug_options) { in PtxOptsFromDebugOptions()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_compile_op_support.cc62 int num_partitions, const DebugOptions* debug_options, const int* seed, in CreateModuleConfig()
141 int num_partitions, const DebugOptions* debug_options) { in CreateModuleConfig()
320 auto debug_options = xla::DebugOptions(); in CreateHloModules() local
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dgraph_mgr.cc105 const DebugOptions& debug_options, Graph* graph, Device* device) { in DecorateAndPublishGraphForDebug()
127 const DebugOptions& debug_options, in InitItem()
299 const DebugOptions& debug_options, in Register()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dtoken_hlo_test.cc142 DebugOptions debug_options = GetDebugOptionsForTest(); in XLA_TEST_F() local
177 DebugOptions debug_options = GetDebugOptionsForTest(); in XLA_TEST_F() local
H A Dhlo_test_base.cc201 auto debug_options = GetDebugOptionsFromFlags(); in GetDebugOptionsForTest() local
541 DebugOptions debug_options = config.debug_options(); in Run() local
635 DebugOptions debug_options = config.debug_options(); in RunMultipleTimes() local
H A Dmatmul_test.cc32 auto debug_options = HloTestBase::GetDebugOptionsForTest(); in GetDebugOptionsForTest() local
/aosp_15_r20/external/tensorflow/tensorflow/core/debug/
H A Dgrpc_session_debug_test.cc169 DebugOptions* debug_options = options.mutable_debug_options(); in TEST_F() local
255 DebugOptions* debug_options = options.mutable_debug_options(); in TEST_F() local
H A Ddebugger_state_impl.h48 DebugGraphDecorator(const DebugOptions& debug_options) in DebugGraphDecorator()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dmark_for_compilation_pass.cc120 MarkForCompilationPassImpl(DebugOptions debug_options, Graph* graph, in MarkForCompilationPassImpl()
1822 const MarkForCompilationPassImpl::DebugOptions& debug_options) { in MarkForCompilation()
1873 MarkForCompilationPassImpl::DebugOptions debug_options; in Run() local
1894 MarkForCompilationPassImpl::DebugOptions debug_options; in RunForTest() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Ddirect_session.cc496 const DebugOptions& debug_options, Graph* graph, Device* device) { in DecorateAndPublishGraphForDebug()
970 DebugOptions debug_options; in PRunSetup() local
1421 const DebugOptions& debug_options = in CreateExecutors() local
1953 DebugOptions debug_options; in RunCallable() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/
H A Dhlo_module_importer.cc63 xla::DebugOptions debug_options; in Import() local

123