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

[INTEL MKL] Removed the macro TF_CALL_DATASET_TYPES as it is redundant

Browse Source
This commit is contained in:
Xiaoming (Jason) Cui 2019-04-30 11:31:46 -07:00
parent a068d4a458
commit 39170eb1a7
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/core/kernels/gather_nd_op_cpu_impl.h
View File

@ -32,8 +32,6 @@ limitations under the License.
#include "tensorflow/core/platform/types.h"
#include "tensorflow/core/util/util.h"
#define TF_CALL_DATASET_TYPES(m) TF_CALL_ALL_TYPES(m) TF_CALL_QUANTIZED_TYPES(m)
namespace tensorflow {
typedef Eigen::ThreadPoolDevice CPUDevice;
@ -153,7 +151,8 @@ struct GatherNdSlice<CPUDevice, T, Index, IXDIM> {
REGISTER_GATHER_ND_FULL(type, int32); \
REGISTER_GATHER_ND_FULL(type, int64)
TF_CALL_DATASET_TYPES(REGISTER_GATHER_ND_CPU);
TF_CALL_ALL_TYPES(REGISTER_GATHER_ND_CPU);
TF_CALL_QUANTIZED_TYPES(REGISTER_GATHER_ND_CPU);
} // namespace functor