diff --git a/libnd4j/include/ops/declarable/headers/datatypes.h b/libnd4j/include/ops/declarable/headers/datatypes.h index d8ff39d48..7c96ae4c7 100644 --- 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 } }