Fixes a race condition in TensorForest tree traversal code resulting in...
Fixes a race condition in TensorForest tree traversal code resulting in use-after-free of input dense/sparse tensors. The race occurs when multiple TreePredictionsV4Op kernels are invoked simultaneously resulting in data_set_.set_input_tensors() being invoked concurrently with tree traversal code accessing the current tensors. PiperOrigin-RevId: 170497611
Loading
Please sign in to comment