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

Making serial execute default in google platform

Browse Source
This commit is contained in:
srinivasan.narayanamoorthy 2019-10-08 13:18:38 -07:00
parent 728cde9e17
commit b5e25a4d7d
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/core/kernels/scatter_functor.h
View File

@ -264,6 +264,12 @@ struct ScatterFunctorBase {
typename TTypes<T>::Matrix params,
typename TTypes<T>::ConstMatrix updates,
typename TTypes<Index>::ConstFlat indices) {
#ifdef PLATFORM_GOOGLE
// The parallel version is significantly slower internally. Only call the
// serial version for now.
// TODO(penporn): Avoid locking in parallelization (sort beforehand).
return SerialExecute(c, d, params, updates, indices);
#else
// indices and params sizes were validated in DoCompute().
const Index N = static_cast<Index>(indices.size());
const Index limit = static_cast<Index>(params.dimension(0));
@ -282,6 +288,7 @@ struct ScatterFunctorBase {
return SerialExecute(c, d, params, updates, indices);
else
return ParallelExecute(c, d, params, updates, indices);
#endif // PLATFORM_GOOGLE
}
};