Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for RegisterRunner (0.14 sec)

  1. tensorflow/cc/training/coordinator_test.cc

      std::unique_ptr<MockQueueRunner> qr1(new MockQueueRunner(&coord));
      qr1->StartCounting(&counter, 100);
      TF_ASSERT_OK(coord.RegisterRunner(std::move(qr1)));
    
      std::unique_ptr<MockQueueRunner> qr2(new MockQueueRunner(&coord));
      qr2->StartCounting(&counter, 100);
      TF_ASSERT_OK(coord.RegisterRunner(std::move(qr2)));
    
      // Wait until the counting has started
      while (counter.load() == 0)
        ;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 23 09:10:43 UTC 2023
    - 6.9K bytes
    - Viewed (0)
  2. tensorflow/cc/training/coordinator.h

    /// thread or any client.
    /// Usage, in the client:
    ///   Coordinator coord;
    ///   std::unique_ptr<QueueRunner> qr(&coord, ...);
    ///   qr.Start(session);
    ///   coord.RegisterRunner(std::move(qr));
    ///   /// do some work
    ///   TF_CHECK_OK(coord.Join());
    /// In each thread of QueueRunner, the coordinator needs to be used as:
    ///   void Run() {
    ///     while (!coord->ShouldStop()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Oct 12 08:49:52 UTC 2023
    - 4.4K bytes
    - Viewed (0)
  3. tensorflow/cc/training/coordinator.cc

          clean_stop_errors_.insert(static_cast<int>(code));
        }
      }
    }
    
    Coordinator::~Coordinator() {
      RequestStop().IgnoreError();
      Join().IgnoreError();
    }
    
    Status Coordinator::RegisterRunner(std::unique_ptr<RunnerInterface> runner) {
      {
        mutex_lock l(mu_);
        if (should_stop_) {
          return Status(absl::StatusCode::kFailedPrecondition,
                        "The coordinator has been stopped.");
        }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 13 08:30:37 UTC 2024
    - 3.2K bytes
    - Viewed (0)
  4. tensorflow/cc/training/queue_runner_test.cc

      std::unique_ptr<QueueRunner> qr1;
      TF_EXPECT_OK(QueueRunner::New(queue_runner1, &coord, &qr1));
      TF_CHECK_OK(qr1->Start(session.get()));
    
      TF_EXPECT_OK(coord.RegisterRunner(std::move(qr0)));
      TF_EXPECT_OK(coord.RegisterRunner(std::move(qr1)));
    
      std::vector<Tensor> dq;
      TF_EXPECT_OK(session->Run({}, {kDequeueOp1}, {}, &dq));
      EXPECT_EQ(*dq[0].scalar<int>().data(), 10);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Sep 21 06:27:51 UTC 2019
    - 14.7K bytes
    - Viewed (0)
Back to top