experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

remove out of date TODO

Browse Source 
PiperOrigin-RevId: 282587527
Change-Id: I39fe00829e92a6f54cdbb37a984f9df1fce7501f
This commit is contained in:
A. Unique TensorFlower 2019-11-26 10:02:34 -08:00 committed by TensorFlower Gardener
parent 9485d6e5ef
commit d5d25ec04e
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/core/kernels/searchsorted_op.cc
View File

@ -37,7 +37,7 @@ struct UpperBoundFunctor<CPUDevice, T, OutType> {
const typename TTypes<T, 1>::ConstTensor& values,
int batch_size, int num_inputs, int num_values,
typename TTypes<OutType, 1>::Tensor* output) {
// TODO(eriche): If anyone ever needs this to be faster, we can multithread.
// TODO(rmlarsen): add multithreading or interleaving.
for (int b = 0; b < batch_size; ++b) {
const T* sorted_inputs_ptr = sorted_inputs.data() + b * num_inputs;
OutType* output_ptr = output->data() + b * num_values;
@ -60,7 +60,7 @@ struct LowerBoundFunctor<CPUDevice, T, OutType> {
const typename TTypes<T, 1>::ConstTensor& values,
int batch_size, int num_inputs, int num_values,
typename TTypes<OutType, 1>::Tensor* output) {
// TODO(eriche): If anyone ever needs this to be faster, we can multithread.
// TODO(rmlarsen): add multithreading or interleaving.
for (int b = 0; b < batch_size; ++b) {
const T* sorted_inputs_ptr = sorted_inputs.data() + b * num_inputs;
OutType* output_ptr = output->data() + b * num_values;