Commit 76e78044 authored by Tayo Oguntebi's avatar Tayo Oguntebi Committed by TensorFlower Gardener
Browse files

Fixes #22750 

  Fixes multi-thread vulnerability in NonMaxSuppressionV3V4 ops.
  Previous implementation used a NonMaxSuppressionV3V4 base class in order to
  reduce code replication. However, this required storing of tensor variables
  in class members, which caused the Compute() method to be non-thread safe.
  This patch removes all op kernel class members, guarateeing thread safety at
  the expense of slightly more code.

PiperOrigin-RevId: 217785064
parent f6566dae
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment