From 863ff76878e7969a98ee20f2ac2d0573d25089d2 Mon Sep 17 00:00:00 2001 From: shugeo Date: Wed, 2 Oct 2019 12:17:00 +0300 Subject: [PATCH] Added declaration for bincast op. --- libnd4j/include/ops/declarable/headers/datatypes.h | 8 ++++++++ 1 file changed, 8 insertions(+) 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 } }