Only use hardware traces on a single subgraph since tracing more than that

isn't supported

PiperOrigin-RevId: 166364134

1 files changed, 3 insertions, 3 deletions

diff --git a/tensorflow/cc/training/queue_runner.h b/tensorflow/cc/training/queue_runner.h
index 71ed44c9c6..2d34500323 100644
--- a/tensorflow/cc/training/queue_runner.h
+++ b/tensorflow/cc/training/queue_runner.h
@@ -62,13 +62,13 @@ class QueueRunner : public RunnerInterface {
   /// Starts the queue runner with the given session and sets the run arguments
   /// for sess->Run. It also collects and stores the cost model.
   Status StartAndCollectCostGraph(Session* sess,
-                                  const RunOptions* run_options = nullptr);
+                                  const RunOptions& run_options = RunOptions());
 
   /// Starts the queue runner with the given session, and wait for up to the
   /// specified time (in milliseconds) for the queues to start to fill up.
   Status Start(Session* sess, int wait_for_ms);
   Status StartAndCollectCostGraph(Session* session, int wait_for_ms,
-                                  const RunOptions* run_options = nullptr);
+                                  const RunOptions& run_options = RunOptions());
 
   /// Requests to stop and runs the cancel op. It would be called in a separate
   /// thread when coordinator is set. If there is no coordinator it should be
@@ -105,7 +105,7 @@ class QueueRunner : public RunnerInterface {
 
   bool IsRunning() const override { return !stopped_; }
 
-  void SetRunArgumentsAndCostGraph(const RunOptions* run_options);
+  void SetRunArgumentsAndCostGraph(const RunOptions& run_options);
 
   Status RealRun(Session* sess, const string& op, bool update_costs);