- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for CostGraphDef (0.66 sec)
-
tensorflow/cc/training/coordinator.h
/// 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 /// its queue.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Oct 12 08:49:52 UTC 2023 - 4.4K bytes - Viewed (0) -
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) -
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) -
tensorflow/cc/training/coordinator.cc
return status_; } 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) -
tensorflow/cc/training/queue_runner_test.cc
TF_EXPECT_OK(session->Run({}, {kDequeueOp0}, {}, &dq0)); // 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) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Sep 21 06:27:51 UTC 2019 - 14.7K bytes - Viewed (0)