@@ -777,12 +777,10 @@ NNVM_REGISTER_ELEMWISE_BINARY_SCALAR(__equal_scalar__)
777
777
" FTVMCompute" , [](const NodeAttrs& attrs,
778
778
const Array<Tensor>& inputs,
779
779
const Array<Tensor>& out_info) {
780
-
781
780
Tensor out = topi::cast (
782
781
binary_scalar_op (attrs, inputs[0 ],
783
782
[](Expr x, Expr y) { return x == y; }),
784
- out_info[0 ]->dtype
785
- );
783
+ out_info[0 ]->dtype );
786
784
return Array<Tensor>{ out };
787
785
})
788
786
.set_attr<nnvm::FGradient>(" FGradient" , MakeZeroGradNodes);
@@ -799,8 +797,7 @@ NNVM_REGISTER_ELEMWISE_BINARY_SCALAR(__not_equal_scalar__)
799
797
Tensor out = topi::cast (
800
798
binary_scalar_op (attrs, inputs[0 ],
801
799
[](Expr x, Expr y) { return x != y; }),
802
- out_info[0 ]->dtype
803
- );
800
+ out_info[0 ]->dtype );
804
801
return Array<Tensor>{ out };
805
802
})
806
803
.set_attr<nnvm::FGradient>(" FGradient" , MakeZeroGradNodes);
@@ -817,8 +814,7 @@ NNVM_REGISTER_ELEMWISE_BINARY_SCALAR(__greater_scalar__)
817
814
Tensor out = topi::cast (
818
815
binary_scalar_op (attrs, inputs[0 ],
819
816
[](Expr x, Expr y) { return x > y; }),
820
- out_info[0 ]->dtype
821
- );
817
+ out_info[0 ]->dtype );
822
818
return Array<Tensor>{ out };
823
819
})
824
820
.set_attr<nnvm::FGradient>(" FGradient" , MakeZeroGradNodes);
@@ -835,8 +831,7 @@ NNVM_REGISTER_ELEMWISE_BINARY_SCALAR(__greater_equal_scalar__)
835
831
Tensor out = topi::cast (
836
832
binary_scalar_op (attrs, inputs[0 ],
837
833
[](Expr x, Expr y) { return x >= y; }),
838
- out_info[0 ]->dtype
839
- );
834
+ out_info[0 ]->dtype );
840
835
return Array<Tensor>{ out };
841
836
})
842
837
.set_attr<nnvm::FGradient>(" FGradient" , MakeZeroGradNodes);
@@ -853,8 +848,7 @@ NNVM_REGISTER_ELEMWISE_BINARY_SCALAR(__less_scalar__)
853
848
Tensor out = topi::cast (
854
849
binary_scalar_op (attrs, inputs[0 ],
855
850
[](Expr x, Expr y) { return x < y; }),
856
- out_info[0 ]->dtype
857
- );
851
+ out_info[0 ]->dtype );
858
852
return Array<Tensor>{ out };
859
853
})
860
854
.set_attr<nnvm::FGradient>(" FGradient" , MakeZeroGradNodes);
@@ -871,8 +865,7 @@ NNVM_REGISTER_ELEMWISE_BINARY_SCALAR(__less_equal_scalar__)
871
865
Tensor out = topi::cast (
872
866
binary_scalar_op (attrs, inputs[0 ],
873
867
[](Expr x, Expr y) { return x <= y; }),
874
- out_info[0 ]->dtype
875
- );
868
+ out_info[0 ]->dtype );
876
869
return Array<Tensor>{ out };
877
870
})
878
871
.set_attr<nnvm::FGradient>(" FGradient" , MakeZeroGradNodes);
0 commit comments