Added declaration for bincast op.

2019-10-02 12:17:00 +03:00 · 2019-10-02 12:17:00 +03:00 · 863ff76878
parent 62326542e9
commit 863ff76878
1 changed files with 8 additions and 0 deletions
--- a/libnd4j/include/ops/declarable/headers/datatypes.h
+++ b/libnd4j/include/ops/declarable/headers/datatypes.h
@ -99,6 +99,14 @@ namespace nd4j {
        #if NOT_EXCLUDED(OP_cast)
        DECLARE_CUSTOM_OP(cast, 1, 1, false, 0, 1);
        #endif
+        /**
+         * This operation change type of input and modified shape of output to conform with given data type
+         *
+         * all as above op
+         * */
+        #if NOT_EXCLUDED(OP_bincast)
+                DECLARE_CUSTOM_OP(bincast, 1, 1, false, 0, 1);
+        #endif
    }
 }