blob: 868204f86eb5f5d8e00e8a74d15d649f8c7f6ab4 (
plain)
1
2
3
4
5
6
7
8
|
#include "tensorflow/core/kernels/cwise_ops_common.h"
namespace tensorflow {
REGISTER2(UnaryOp, CPU, "IsInf", functor::isinf, float, double);
#if GOOGLE_CUDA
REGISTER2(UnaryOp, GPU, "IsInf", functor::isinf, float, double);
#endif
} // namespace tensorflow
|