Home
last modified time | relevance | path

Searched refs:profile_steps (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/python/profiler/
Dprofile_context.py215 def add_auto_profiling(self, cmd, options, profile_steps): argument
227 self._auto_profiles.append((cmd, options, profile_steps[:]))
228 self._slow_path_steps |= set(profile_steps)
229 self._trace_steps |= set(profile_steps)
Dprofile_context_test.py51 pctx.add_auto_profiling("op", options=opts, profile_steps=[15, 50, 100])