Skip to content

Commit

Permalink
Merge pull request #42682 from kpedro88/revertDeepMET
Browse files Browse the repository at this point in the history
revert DeepMET TensorFlow workaround
  • Loading branch information
cmsbuild authored Sep 1, 2023
2 parents 6bcf8af + 1842a69 commit d009690
Showing 1 changed file with 2 additions and 11 deletions.
13 changes: 2 additions & 11 deletions RecoMET/METPUSubtraction/plugins/DeepMETProducer.cc
Original file line number Diff line number Diff line change
Expand Up @@ -43,17 +43,8 @@ DeepMETProducer::DeepMETProducer(const edm::ParameterSet& cfg, const tensorflow:
session_(cache->getSession()) {
produces<pat::METCollection>();

// Workaround for missing constructor TensorShape::TensorShape(absl::Slice<long>),
// the constructor expects Slice<int64> or initializer_list<int64> and is marked explicit
tensorflow::TensorShape shape;
shape.AddDim(1);
shape.AddDim(max_n_pf_);
shape.AddDim(8);

tensorflow::TensorShape cat_shape;
cat_shape.AddDim(1);
cat_shape.AddDim(max_n_pf_);
cat_shape.AddDim(1);
const tensorflow::TensorShape shape({1, max_n_pf_, 8});
const tensorflow::TensorShape cat_shape({1, max_n_pf_, 1});

input_ = tensorflow::Tensor(tensorflow::DT_FLOAT, shape);
input_cat0_ = tensorflow::Tensor(tensorflow::DT_FLOAT, cat_shape);
Expand Down

0 comments on commit d009690

Please sign in to comment.