Brutex
/
cavis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added declaration for bincast op.

master
shugeo 2019-10-02 12:17:00 +03:00
parent 62326542e9
commit 863ff76878
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
libnd4j/include/ops/declarable/headers/datatypes.h
View File

@ -99,6 +99,14 @@ namespace nd4j {
#if NOT_EXCLUDED(OP_cast) #if NOT_EXCLUDED(OP_cast)
DECLARE_CUSTOM_OP(cast, 1, 1, false, 0, 1); DECLARE_CUSTOM_OP(cast, 1, 1, false, 0, 1);
#endif #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
} }
} }