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

minor build fix (#139) 

Signed-off-by: raver119 <raver119@gmail.com>
master
raver119 2019-12-21 08:07:13 +03:00 committed by GitHub
parent 5d9b2a16e5
commit 495256c827
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
libnd4j/include/ops/declarable/generic/transforms/hashcode.cpp
View File

@ -27,7 +27,7 @@
namespace nd4j {
namespace ops {
REDUCTION_OP_IMPL(hashcode, 1, 1, false, 0, 0) {
CUSTOM_OP_IMPL(hashcode, 1, 1, false, 0, 0) {
REQUIRE_TRUE(block.width() == 1, 0, "hashcode: this op can't be applied along dimension");
auto input = INPUT_VARIABLE(0);
@ -40,6 +40,10 @@ namespace nd4j {
return Status::OK();
};
DECLARE_SHAPE_FN(hashcode) {
return SHAPELIST(ConstantShapeHelper::getInstance()->scalarShapeInfo(nd4j::DataType::INT64));
}
DECLARE_TYPES(hashcode) {
getOpDescriptor()

2
libnd4j/include/ops/declarable/headers/transforms.h
View File

@ -213,7 +213,7 @@ namespace nd4j {
* This operation calculates hash code, optionally along dimension
*/
#if NOT_EXCLUDED(OP_hashcode)
DECLARE_REDUCTION_OP(hashcode, 1, 1, false, 0, 0);
DECLARE_CUSTOM_OP(hashcode, 1, 1, false, 0, 0);
#endif
/**