blob: 17f28340305bb7e5610d029bbd7a13a182cf6f43 (
plain)
1
2
3
4
5
6
|
#include "tensorflow/core/kernels/cwise_ops_common.h"
namespace tensorflow {
REGISTER2(BinaryOp, CPU, "Mod", functor::mod, int32, int64);
REGISTER2(BinaryOp, CPU, "Mod", functor::fmod, float, double);
} // namespace tensorflow
|