提交 a5cca4d3 编写于 作者: Tim Khatkevich's avatar Tim Khatkevich 提交者: Facebook Github Bot

add failback for Sign operator (#21343)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/21343

Needed to binarise features

Reviewed By: yinghai

Differential Revision: D15625653

fbshipit-source-id: 52f48259a040dac35a7000bb1eea9feb5c7ef1ab
上级 51fb42eb
......@@ -199,6 +199,12 @@ REGISTER_IDEEP_OPERATOR(
Sqrt,
IDEEPFallbackOp<UnaryElementwiseOp<
TensorTypes<float>, CPUContext, SqrtFunctor<CPUContext>>>);
REGISTER_IDEEP_OPERATOR(
Sign,
IDEEPFallbackOp<UnaryElementwiseOp<
TensorTypes<float>,
CPUContext,
SignFunctor<CPUContext>>>);
REGISTER_IDEEP_OPERATOR(
Div,
IDEEPFallbackOp<BinaryElementwiseOp<
......
Markdown 格式
0% or
您添加了 0 到此讨论。请谨慎行事。
先完成此消息的编辑!
想要评论请 注册