Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for ExportCostGraph (0.15 sec)

  1. tensorflow/cc/training/coordinator.h

    /// The abstract interface for runners which must implement the Join and the
    /// IsRunning function.
    class RunnerInterface {
     public:
      virtual ~RunnerInterface() {}
      virtual Status Join() = 0;
      virtual Status ExportCostGraph(CostGraphDef* cost_graph) const {
        return Status(absl::StatusCode::kInvalidArgument,
                      "No cost model to export.");
      }
      /// Returns true iff the runner is running, i.e. if it is trying to populate
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Oct 12 08:49:52 UTC 2023
    - 4.4K bytes
    - Viewed (0)
  2. tensorflow/cc/training/coordinator.cc

    void Coordinator::WaitForStop() {
      mutex_lock l(mu_);
      while (!should_stop_) {
        wait_for_stop_.wait(l);
      }
    }
    
    Status Coordinator::ExportCostGraph(CostGraphDef* cost_graph) const {
      mutex_lock l(runners_lock_);
      for (auto& t : runners_) {
        Status s = t->ExportCostGraph(cost_graph);
        if (!s.ok()) {
          return s;
        }
      }
      return absl::OkStatus();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 13 08:30:37 UTC 2024
    - 3.2K bytes
    - Viewed (0)
  3. tensorflow/cc/training/queue_runner.h

      /// otherwise returns the first captured failure status.
      Status Join() final;
    
      /// Returns the latest status.
      Status GetStatus();
    
      // Returns the stored cost model.
      Status ExportCostGraph(CostGraphDef* cost_graph) const override;
    
     private:
      QueueRunner() : coord_(nullptr), stopped_(false), cg_mu_(nullptr) {}
    
      // Initializes the instance with the QueueRunnerDef proto.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Aug 31 18:27:00 UTC 2022
    - 5K bytes
    - Viewed (0)
  4. tensorflow/cc/training/queue_runner_test.cc

      // Second call to run dequeue op is to make sure the cost graph has been
      // stored.
      TF_EXPECT_OK(session->Run({}, {kDequeueOp0}, {}, &dq0));
    
      CostGraphDef cost_graph;
      TF_CHECK_OK(qr->ExportCostGraph(&cost_graph));
      EXPECT_TRUE(cost_graph.node_size() > 0);
    
      qr->Stop(session.get());
    }
    
    TEST(QueueRunnerTest, NoRunMetaDataTest) {
      GraphDef graph_def = BuildSimpleGraph();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Sep 21 06:27:51 UTC 2019
    - 14.7K bytes
    - Viewed (0)
  5. tensorflow/cc/training/queue_runner.cc

        if (coord_) {
          coord_->RequestStop().IgnoreError();
        }
      }
    }
    
    Status QueueRunner::GetStatus() {
      mutex_lock l(mu_);
      return status_;
    }
    
    Status QueueRunner::ExportCostGraph(CostGraphDef* cost_graph) const {
      if (!cg_mu_) {
        return Status(absl::StatusCode::kFailedPrecondition,
                      "This QueueRunner doesn't collect a cost graph.");
      }
      mutex_lock l(*cg_mu_);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 13 08:30:37 UTC 2024
    - 7K bytes
    - Viewed (0)
Back to top