Skip to content

Commit

Permalink
Delete batch_schedulers_ before thread_pool_name_
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 439229205
  • Loading branch information
tensorflower-gardener authored and tensorflow-copybara committed Apr 4, 2022
1 parent 48a699a commit 2837812
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions tensorflow_serving/batching/batching_session.cc
Original file line number Diff line number Diff line change
Expand Up @@ -260,15 +260,15 @@ class BatchingSession : public ServingSession {
std::unique_ptr<Batch<BatchingSessionTask>> batch);

const BatchingSessionOptions options_;
// The name of the thread pool of the underlying batch scheduler. It is used
// for monitoring purpose, and can be empty if not known.
const std::string thread_pool_name_;

std::unique_ptr<Session> wrapped_;
std::unordered_map<TensorSignature,
std::unique_ptr<BatchScheduler<BatchingSessionTask>>,
HashTensorSignature, EqTensorSignature>
batch_schedulers_;
// The name of the thread pool of the underlying batch scheduler. It is used
// for monitoring purpose, and can be empty if not known.
const std::string thread_pool_name_;

// If set, default_scheduler_creator_ is used when the input signature does
// not match any existing signature defined during model load. This helps
Expand Down

0 comments on commit 2837812

Please sign in to comment.