Skip to content

Commit

Permalink
undoing changes to network_tf_cc.cc that should have been done on sea…
Browse files Browse the repository at this point in the history
…rch.cc
  • Loading branch information
KarlKfoury authored Aug 30, 2024
1 parent a6b5d3a commit b12e68b
Showing 1 changed file with 0 additions and 6 deletions.
6 changes: 0 additions & 6 deletions src/neural/network_tf_cc.cc
Original file line number Diff line number Diff line change
Expand Up @@ -343,15 +343,9 @@ class TFNetworkComputation : public NetworkComputation {
raw_input_.emplace_back(input);
}
void ComputeBlocking() override {
auto start_time = std::chrono::high_resolution_clock::now();
PrepareInput();
status_ = network_->Compute(input_, &output_);
CHECK(status_.ok()) << status_.ToString();
auto end_time = std::chrono::high_resolution_clock::now();
std::chrono::duration<double> elapsed = end_time - start_time;
if (elapsed.count() > 1.0) {
std::cerr << "Warning: Computation took " << elapsed.count() << " seconds, which is longer than expected." << std::endl;
}
}

int GetBatchSize() const override { return raw_input_.size(); }
Expand Down

0 comments on commit b12e68b

Please sign in to comment.