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

build fix 

Signed-off-by: raver119 <raver119@gmail.com>
master
raver119 2019-08-03 09:15:12 +03:00 committed by AlexDBlack
parent 3c4e959e21
commit e2b7675e9c
2 changed files with 14 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
libnd4j/include/ops/declarable/helpers/cuda/gradient.cu
View File

@ -25,7 +25,7 @@ namespace nd4j {
namespace ops { namespace ops {
namespace helpers { namespace helpers {
template <typename T> template <typename T>
static void applyGradientDescent_(LaunchContext* context, NDArray* input, NDArray* step, double weight, NDArray* output) { void applyGradientDescent_(LaunchContext* context, NDArray* input, NDArray* step, double weight, NDArray* output) {
auto lambda = LAMBDA_TT(_x, _y, weight) { auto lambda = LAMBDA_TT(_x, _y, weight) {
return _x - (_y * weight); return _x - (_y * weight);
}; };

20
nd4j/nd4j-backends/nd4j-backend-impls/nd4j-native/src/main/java/org/nd4j/nativeblas/Nd4jCpu.java
View File

@ -3814,11 +3814,9 @@ public native @Cast("char*") String runFullBenchmarkSuit(@Cast("bool") boolean p
public native void syncToDevice(); public native void syncToDevice();
public native void syncShape(); public native void syncShape();
/** // #ifndef __JAVACPP_HACK__
* This method can be used on architectures that use special buffers
* @param writeList // #endif
* @param readList
*/
/** /**
* This method returns buffer pointer offset by given number of elements, wrt own data type * This method returns buffer pointer offset by given number of elements, wrt own data type
@ -4056,6 +4054,7 @@ public native @Cast("char*") String runFullBenchmarkSuit(@Cast("bool") boolean p
/** /**
* this method assigns values of given array to this one * this method assigns values of given array to this one
*/ */
public native void assign(@Const NDArray other, @Cast("bool") boolean allowParallelism/*=true*/);
public native void assign(@Const NDArray other); public native void assign(@Const NDArray other);
/** /**
@ -4065,12 +4064,19 @@ public native @Cast("char*") String runFullBenchmarkSuit(@Cast("bool") boolean p
/** /**
* this method assigns given value to all elements in array * this method assigns given value to all elements in array
*/ */
public native void assign(double value, @Cast("bool") boolean allowParallelism/*=true*/);
public native void assign(double value); public native void assign(double value);
public native void assign(float value, @Cast("bool") boolean allowParallelism/*=true*/);
public native void assign(float value); public native void assign(float value);
public native void assign(@Cast("const float16") short value, @Cast("bool") boolean allowParallelism/*=true*/);
public native void assign(@Cast("const float16") short value); public native void assign(@Cast("const float16") short value);
public native void assign(@Cast("const Nd4jLong") long value, @Cast("bool") boolean allowParallelism/*=true*/);
public native void assign(@Cast("const Nd4jLong") long value); public native void assign(@Cast("const Nd4jLong") long value);
public native void assign(int value, @Cast("bool") boolean allowParallelism/*=true*/);
public native void assign(int value); public native void assign(int value);
public native void assign(@Cast("const uint8_t") byte value, @Cast("bool") boolean allowParallelism/*=true*/);
public native void assign(@Cast("const uint8_t") byte value); public native void assign(@Cast("const uint8_t") byte value);
public native void assign(@Cast("const bool") boolean value, @Cast("bool") boolean allowParallelism/*=true*/);
public native void assign(@Cast("const bool") boolean value); public native void assign(@Cast("const bool") boolean value);
/** /**
@ -20153,7 +20159,7 @@ public static final int TAD_THRESHOLD = TAD_THRESHOLD();
* CAUTION: either size tensor or a pair of int params should be provided. * CAUTION: either size tensor or a pair of int params should be provided.
*/ */
// #if NOT_EXCLUDED(OP_resize_bilinear) // #if NOT_EXCLUDED(OP_resize_nearest_neighbor)
@Namespace("nd4j::ops") public static class resize_nearest_neighbor extends DeclarableCustomOp { @Namespace("nd4j::ops") public static class resize_nearest_neighbor extends DeclarableCustomOp {
static { Loader.load(); } static { Loader.load(); }
/** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */ /** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */
@ -20982,7 +20988,7 @@ public static final int TAD_THRESHOLD = TAD_THRESHOLD();
public native ShapeList calculateOutputShape(ShapeList inputShape, @ByRef Context block); public native ShapeList calculateOutputShape(ShapeList inputShape, @ByRef Context block);
} }
// #endif // #endif
// #if NOT_EXCLUDED(OP_batchnorm) // #if NOT_EXCLUDED(OP_batchnorm_new)
@Namespace("nd4j::ops") public static class batchnorm_new extends DeclarableCustomOp { @Namespace("nd4j::ops") public static class batchnorm_new extends DeclarableCustomOp {
static { Loader.load(); } static { Loader.load(); }
/** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */ /** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */