Skip to content
Permalink
Browse files

[WIP] No more axis (eclipse#6902)

Axis overhaul
  • Loading branch information...
raver119 authored and printomi committed Jan 7, 2019
1 parent ea34331 commit b69b83e117ce4611c921e92aa866b55ef82d3c3a
Showing with 1,613 additions and 23,627 deletions.
  1. +1 −1 ...earning4j/deeplearning4j-nn/src/main/java/org/deeplearning4j/nn/conf/dropout/GaussianDropout.java
  2. +1 −2 ...4j/deeplearning4j-nn/src/main/java/org/deeplearning4j/nn/graph/vertex/impl/ElementWiseVertex.java
  3. +1 −1 ...4j/deeplearning4j-nn/src/main/java/org/deeplearning4j/nn/graph/vertex/impl/L2NormalizeVertex.java
  4. +2 −2 deeplearning4j/deeplearning4j-nn/src/main/java/org/deeplearning4j/nn/graph/vertex/impl/L2Vertex.java
  5. +1 −1 ...ing4j-nn/src/main/java/org/deeplearning4j/nn/layers/convolution/subsampling/SubsamplingLayer.java
  6. +7 −10 ...eeplearning4j-nn/src/main/java/org/deeplearning4j/nn/layers/normalization/BatchNormalization.java
  7. +2 −2 ...ng4j/deeplearning4j-nn/src/main/java/org/deeplearning4j/nn/layers/pooling/GlobalPoolingLayer.java
  8. +3 −5 ...earning4j/deeplearning4j-nn/src/main/java/org/deeplearning4j/nn/layers/recurrent/LSTMHelpers.java
  9. +1 −2 ...ng4j/deeplearning4j-nn/src/main/java/org/deeplearning4j/optimize/solvers/BackTrackLineSearch.java
  10. +4 −4 deeplearning4j/deeplearning4j-nn/src/main/java/org/deeplearning4j/util/MaskedReductionUtil.java
  11. +24 −16 libnd4j/blas/NativeOps.h
  12. +14 −5 libnd4j/blas/cpu/NativeOpExcutioner.cpp
  13. +55 −18 libnd4j/blas/cpu/NativeOps.cpp
  14. +67 −23 libnd4j/blas/cuda/NativeOps.cu
  15. +0 −176 libnd4j/include/b40c/partition/downsweep/cta.cuh
  16. +0 −147 libnd4j/include/b40c/partition/downsweep/kernel_policy.cuh
  17. +0 −1,158 libnd4j/include/b40c/partition/downsweep/tile.cuh
  18. +0 −129 libnd4j/include/b40c/partition/downsweep/tuning_policy.cuh
  19. +0 −123 libnd4j/include/b40c/partition/policy.cuh
  20. +0 −65 libnd4j/include/b40c/partition/problem_type.cuh
  21. +0 −72 libnd4j/include/b40c/partition/spine/kernel.cuh
  22. +0 −220 libnd4j/include/b40c/partition/upsweep/aggregate_counters.cuh
  23. +0 −102 libnd4j/include/b40c/partition/upsweep/composite_counters.cuh
  24. +0 −315 libnd4j/include/b40c/partition/upsweep/cta.cuh
  25. +0 −130 libnd4j/include/b40c/partition/upsweep/kernel_policy.cuh
  26. +0 −201 libnd4j/include/b40c/partition/upsweep/tile.cuh
  27. +0 −97 libnd4j/include/b40c/partition/upsweep/tuning_policy.cuh
  28. +0 −659 libnd4j/include/b40c/radix_sort/autotuned_policy.cuh
  29. +0 −102 libnd4j/include/b40c/radix_sort/downsweep/cta.cuh
  30. +0 −264 libnd4j/include/b40c/radix_sort/downsweep/kernel.cuh
  31. +0 −61 libnd4j/include/b40c/radix_sort/downsweep/kernel_policy.cuh
  32. +0 −147 libnd4j/include/b40c/radix_sort/downsweep/tile.cuh
  33. +0 −93 libnd4j/include/b40c/radix_sort/downsweep/tuning_policy.cuh
  34. +0 −840 libnd4j/include/b40c/radix_sort/enactor.cuh
  35. +0 −61 libnd4j/include/b40c/radix_sort/pass_policy.cuh
  36. +0 −192 libnd4j/include/b40c/radix_sort/policy.cuh
  37. +0 −67 libnd4j/include/b40c/radix_sort/problem_type.cuh
  38. +0 −190 libnd4j/include/b40c/radix_sort/sort_utils.cuh
  39. +0 −88 libnd4j/include/b40c/radix_sort/upsweep/cta.cuh
  40. +0 −113 libnd4j/include/b40c/radix_sort/upsweep/kernel.cuh
  41. +0 −61 libnd4j/include/b40c/radix_sort/upsweep/kernel_policy.cuh
  42. +0 −132 libnd4j/include/b40c/radix_sort/upsweep/tile.cuh
  43. +0 −86 libnd4j/include/b40c/radix_sort/upsweep/tuning_policy.cuh
  44. +0 −553 libnd4j/include/b40c/reduction/autotuned_policy.cuh
  45. +0 −264 libnd4j/include/b40c/reduction/cta.cuh
  46. +0 −500 libnd4j/include/b40c/reduction/enactor.cuh
  47. +0 −136 libnd4j/include/b40c/reduction/kernel_policy.cuh
  48. +0 −211 libnd4j/include/b40c/reduction/policy.cuh
  49. +0 −61 libnd4j/include/b40c/reduction/problem_type.cuh
  50. +0 −112 libnd4j/include/b40c/reduction/spine/kernel.cuh
  51. +0 −205 libnd4j/include/b40c/reduction/upsweep/kernel.cuh
  52. +0 −641 libnd4j/include/b40c/scan/autotuned_policy.cuh
  53. +0 −203 libnd4j/include/b40c/scan/downsweep/cta.cuh
  54. +0 −116 libnd4j/include/b40c/scan/downsweep/kernel.cuh
  55. +0 −561 libnd4j/include/b40c/scan/enactor.cuh
  56. +0 −161 libnd4j/include/b40c/scan/kernel_policy.cuh
  57. +0 −255 libnd4j/include/b40c/scan/policy.cuh
  58. +0 −65 libnd4j/include/b40c/scan/problem_type.cuh
  59. +0 −113 libnd4j/include/b40c/scan/spine/kernel.cuh
  60. +0 −190 libnd4j/include/b40c/scan/upsweep/cta.cuh
  61. +0 −163 libnd4j/include/b40c/scan/upsweep/kernel.cuh
  62. +0 −96 libnd4j/include/b40c/util/arch_dispatch.cuh
  63. +0 −240 libnd4j/include/b40c/util/basic_utils.cuh
  64. +0 −195 libnd4j/include/b40c/util/cta_work_distribution.cuh
  65. +0 −410 libnd4j/include/b40c/util/cta_work_progress.cuh
  66. +0 −192 libnd4j/include/b40c/util/cuda_properties.cuh
  67. +0 −216 libnd4j/include/b40c/util/device_intrinsics.cuh
  68. +0 −583 libnd4j/include/b40c/util/enactor_base.cuh
  69. +0 −108 libnd4j/include/b40c/util/error_utils.cuh
  70. +0 −213 libnd4j/include/b40c/util/global_barrier.cuh
  71. +0 −158 libnd4j/include/b40c/util/io/gather_tile.cuh
  72. +0 −300 libnd4j/include/b40c/util/io/initialize_tile.cuh
  73. +0 −444 libnd4j/include/b40c/util/io/load_tile.cuh
  74. +0 −488 libnd4j/include/b40c/util/io/load_tile_discontinuity.cuh
  75. +0 −262 libnd4j/include/b40c/util/io/modified_load.cuh
  76. +0 −253 libnd4j/include/b40c/util/io/modified_store.cuh
  77. +0 −340 libnd4j/include/b40c/util/io/scatter_tile.cuh
  78. +0 −229 libnd4j/include/b40c/util/io/store_tile.cuh
  79. +0 −129 libnd4j/include/b40c/util/io/two_phase_scatter_tile.cuh
  80. +0 −316 libnd4j/include/b40c/util/kernel_runtime_stats.cuh
  81. +0 −52 libnd4j/include/b40c/util/memset_kernel.cuh
  82. +0 −170 libnd4j/include/b40c/util/multiple_buffering.cuh
  83. +0 −79 libnd4j/include/b40c/util/numeric_traits.cuh
  84. +0 −92 libnd4j/include/b40c/util/operators.cuh
  85. +0 −196 libnd4j/include/b40c/util/parameter_generation.cuh
  86. +0 −75 libnd4j/include/b40c/util/ping_pong_storage.cuh
  87. +0 −279 libnd4j/include/b40c/util/raking_grid.cuh
  88. +0 −91 libnd4j/include/b40c/util/random_bits.cuh
  89. +0 −287 libnd4j/include/b40c/util/reduction/cooperative_reduction.cuh
  90. +0 −152 libnd4j/include/b40c/util/reduction/serial_reduce.cuh
  91. +0 −256 libnd4j/include/b40c/util/reduction/soa/cooperative_soa_reduction.cuh
  92. +0 −224 libnd4j/include/b40c/util/reduction/soa/serial_soa_reduce.cuh
  93. +0 −179 libnd4j/include/b40c/util/reduction/soa/warp_soa_reduce.cuh
  94. +0 −310 libnd4j/include/b40c/util/reduction/tree_reduce.cuh
  95. +0 −186 libnd4j/include/b40c/util/reduction/warp_reduce.cuh
  96. +0 −510 libnd4j/include/b40c/util/scan/cooperative_scan.cuh
  97. +0 −146 libnd4j/include/b40c/util/scan/serial_scan.cuh
  98. +0 −413 libnd4j/include/b40c/util/scan/soa/cooperative_soa_scan.cuh
  99. +0 −242 libnd4j/include/b40c/util/scan/soa/serial_soa_scan.cuh
  100. +0 −208 libnd4j/include/b40c/util/scan/soa/warp_soa_scan.cuh
  101. +0 −223 libnd4j/include/b40c/util/scan/warp_scan.cuh
  102. +0 −332 libnd4j/include/b40c/util/soa_tuple.cuh
  103. +0 −240 libnd4j/include/b40c/util/spine.cuh
  104. +0 −293 libnd4j/include/b40c/util/srts_details.cuh
  105. +0 −294 libnd4j/include/b40c/util/srts_grid.cuh
  106. +0 −308 libnd4j/include/b40c/util/srts_soa_details.cuh
  107. +0 −124 libnd4j/include/b40c/util/vector_types.cuh
  108. +2 −2 libnd4j/tests_cpu/layers_tests/JavaInteropTests.cpp
  109. +51 −0 libnd4j/tests_cpu/layers_tests/PlaygroundTests.cpp
  110. +5 −5 nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/autodiff/samediff/SameDiff.java
  111. +34 −32 ...ends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/autodiff/samediff/SameDiffOpExecutioner.java
  112. +1 −1 ...4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/autodiff/validation/OpValidation.java
  113. +1 −1 ...ackends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/evaluation/classification/Evaluation.java
  114. +2 −2 ...j-api-parent/nd4j-api/src/main/java/org/nd4j/evaluation/classification/EvaluationCalibration.java
  115. +4 −6 .../nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/evaluation/classification/ROC.java
  116. +1 −1 ...s/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/evaluation/regression/RegressionEvaluation.java
  117. +1 −1 ...kends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/activations/impl/ActivationCube.java
  118. +3 −3 ...ckends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/activations/impl/ActivationELU.java
  119. +1 −1 ...d4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/activations/impl/ActivationHardSigmoid.java
  120. +1 −1 ...s/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/activations/impl/ActivationHardTanH.java
  121. +1 −1 ...ends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/activations/impl/ActivationLReLU.java
  122. +2 −2 ...ends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/activations/impl/ActivationPReLU.java
  123. +1 −1 ...ends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/activations/impl/ActivationRReLU.java
  124. +1 −1 ...4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/activations/impl/ActivationRationalTanh.java
  125. +1 −1 ...kends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/activations/impl/ActivationReLU.java
  126. +1 −1 ...ends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/activations/impl/ActivationReLU6.java
  127. +1 −1 ...j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/activations/impl/ActivationRectifiedTanh.java
  128. +1 −1 ...kends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/activations/impl/ActivationSELU.java
  129. +1 −1 ...ds/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/activations/impl/ActivationSigmoid.java
  130. +1 −1 ...s/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/activations/impl/ActivationSoftPlus.java
  131. +1 −1 ...s/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/activations/impl/ActivationSoftSign.java
  132. +1 −1 ...ds/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/activations/impl/ActivationSoftmax.java
  133. +1 −1 ...ends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/activations/impl/ActivationSwish.java
  134. +1 −1 ...kends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/activations/impl/ActivationTanH.java
  135. +2 −2 ...api-parent/nd4j-api/src/main/java/org/nd4j/linalg/activations/impl/ActivationThresholdedReLU.java
  136. +68 −60 ...nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ndarray/BaseNDArray.java
  137. +4 −4 ...ackends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ndarray/BaseSparseNDArray.java
  138. +12 −0 nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ndarray/INDArray.java
  139. +1 −1 ...-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/BaseBroadcastBoolOp.java
  140. +1 −3 ...nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/BaseBroadcastOp.java
  141. +6 −57 ...ackends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/BaseIndexAccumulation.java
  142. +38 −0 nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/BaseOp.java
  143. +4 −8 ...d4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/BaseReduceBoolOp.java
  144. +8 −8 ...4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/BaseReduceFloatOp.java
  145. +4 −8 ...d4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/BaseReduceLongOp.java
  146. +19 −50 nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/BaseReduceOp.java
  147. +4 −8 ...d4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/BaseReduceSameOp.java
  148. +1 −3 ...d4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/BaseScalarBoolOp.java
  149. +1 −1 nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/BaseScalarOp.java
  150. +4 −0 nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/BroadcastOp.java
  151. +14 −27 ...4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/IndexAccumulation.java
  152. +1 −1 nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/ReduceBoolOp.java
  153. +1 −1 nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/ReduceFloatOp.java
  154. +1 −1 nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/ReduceLongOp.java
  155. +13 −21 nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/ReduceOp.java
  156. +1 −1 nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/ReduceSameOp.java
  157. +19 −2 nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/ScalarOp.java
  158. +48 −68 ...j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/executioner/DefaultOpExecutioner.java
  159. +22 −23 ...nds/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/executioner/OpExecutioner.java
  160. +2 −2 ...nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/executioner/OpExecutionerUtil.java
  161. +4 −4 ...ends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/factory/DefaultOpFactory.java
  162. +4 −19 ...ds/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/indexaccum/FirstIndex.java
  163. +4 −26 ...ackends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/indexaccum/IAMax.java
  164. +4 −23 ...ackends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/indexaccum/IAMin.java
  165. +4 −20 ...backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/indexaccum/IMax.java
  166. +4 −25 ...backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/indexaccum/IMin.java
  167. +4 −10 ...nds/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/indexaccum/LastIndex.java
  168. +4 −8 .../nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/reduce/custom/LogSumExp.java
  169. +3 −7 ...ds/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/reduce/floating/AMean.java
  170. +2 −2 ...nds/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/reduce/floating/Bias.java
  171. +4 −8 .../nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/reduce/floating/Entropy.java
  172. +4 −9 ...4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/reduce/floating/LogEntropy.java
  173. +5 −9 ...nds/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/reduce/floating/Mean.java
  174. +4 −8 ...ds/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/reduce/floating/Norm1.java
  175. +4 −8 ...ds/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/reduce/floating/Norm2.java
  176. +4 −8 .../nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/reduce/floating/NormMax.java
  177. +4 −7 ...pi-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/reduce/floating/ShannonEntropy.java
  178. +2 −2 ...4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/reduce/longer/CountNonZero.java
  179. +2 −2 .../nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/reduce/longer/CountZero.java
  180. +6 −3 ...-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/reduce/longer/MatchCondition.java
  181. +4 −8 ...ackends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/reduce/same/AMax.java
  182. +4 −24 ...ackends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/reduce/same/AMin.java
  183. +4 −8 ...ackends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/reduce/same/ASum.java
  184. +5 −25 ...backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/reduce/same/Max.java
  185. +5 −24 ...backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/reduce/same/Min.java
  186. +5 −25 ...ackends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/reduce/same/Prod.java
  187. +5 −9 ...backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/reduce/same/Sum.java
  188. +10 −21 ...s/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/reduce3/CosineDistance.java
  189. +10 −21 ...nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/reduce3/CosineSimilarity.java
  190. +6 −10 ...d4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/reduce3/Dot.java
  191. +6 −8 ...ds/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/reduce3/EqualsWithEps.java
  192. +12 −14 ...d4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/reduce3/EuclideanDistance.java
  193. +9 −26 .../nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/reduce3/HammingDistance.java
  194. +7 −23 .../nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/reduce3/JaccardDistance.java
  195. +9 −20 ...d4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/reduce3/ManhattanDistance.java
  196. +6 −13 ...pi-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/summarystats/StandardDeviation.java
  197. +14 −34 ...ds/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/summarystats/Variance.java
  198. +1 −1 ...nt/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/transforms/gradient/SoftMaxDerivative.java
  199. +7 −8 ...-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/transforms/strict/OldSoftMax.java
  200. +0 −4 ...rent/nd4j-api/src/main/java/org/nd4j/linalg/api/ops/impl/transforms/strict/SoftMaxDerivative.java
  201. +2 −1 nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/shape/Shape.java
  202. +6 −6 ...nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/convolution/Convolution.java
  203. +4 −3 ...rent/nd4j-api/src/main/java/org/nd4j/linalg/dataset/api/preprocessor/stats/DistributionStats.java
  204. +34 −34 nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/factory/Broadcast.java
  205. +13 −21 nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/factory/Nd4j.java
  206. +13 −12 ...d4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/indexing/BooleanIndexing.java
  207. +1 −1 ...nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/learning/AMSGradUpdater.java
  208. +4 −4 ...nds/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/lossfunctions/impl/LossBinaryXENT.java
  209. +1 −1 ...4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/lossfunctions/impl/LossL1.java
  210. +2 −2 ...-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/lossfunctions/impl/LossMAPE.java
  211. +1 −1 ...nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/lossfunctions/impl/LossMixtureDensity.java
  212. +59 −65 ...4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/ops/transforms/Transforms.java
  213. +1 −1 nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/profiler/OpProfiler.java
  214. +24 −24 nd4j/nd4j-backends/nd4j-api-parent/nd4j-native-api/src/main/java/org/nd4j/nativeblas/NativeOps.java
  215. +3 −0 ...ends/nd4j-backend-impls/nd4j-cuda/src/main/java/org/nd4j/jita/allocator/impl/AtomicAllocator.java
  216. +3 −3 ...ckends/nd4j-backend-impls/nd4j-cuda/src/main/java/org/nd4j/linalg/jcublas/blas/JcublasLevel1.java
  217. +156 −154 ...ackend-impls/nd4j-cuda/src/main/java/org/nd4j/linalg/jcublas/ops/executioner/CudaExecutioner.java
  218. +8 −26 ...nd-impls/nd4j-cuda/src/main/java/org/nd4j/linalg/jcublas/ops/executioner/CudaGridExecutioner.java
  219. +72 −48 nd4j/nd4j-backends/nd4j-backend-impls/nd4j-cuda/src/main/java/org/nd4j/nativeblas/Nd4jCuda.java
  220. +1 −1 ...ckend-impls/nd4j-native/src/main/java/org/nd4j/linalg/cpu/nativecpu/compression/CpuThreshold.java
  221. +79 −120 ...ackend-impls/nd4j-native/src/main/java/org/nd4j/linalg/cpu/nativecpu/ops/NativeOpExecutioner.java
  222. +72 −48 nd4j/nd4j-backends/nd4j-backend-impls/nd4j-native/src/main/java/org/nd4j/nativeblas/Nd4jCpu.java
  223. +8 −8 nd4j/nd4j-backends/nd4j-tests/pom.xml
  224. +1 −1 nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/autodiff/opvalidation/LayerOpValidation.java
  225. +17 −17 .../nd4j-backends/nd4j-tests/src/test/java/org/nd4j/autodiff/opvalidation/ReductionOpValidation.java
  226. +17 −17 .../nd4j-backends/nd4j-tests/src/test/java/org/nd4j/autodiff/opvalidation/TransformOpValidation.java
  227. +5 −5 nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/autodiff/samediff/SameDiffTests.java
  228. +2 −2 nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/imports/TFGraphs/TFGraphTestAllHelper.java
  229. +1 −1 nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/LoneTest.java
  230. +1 −1 nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/NDArrayTestsFortran.java
  231. +60 −60 nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/Nd4jTestsC.java
  232. +1 −1 nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/convolution/ConvolutionTestsC.java
  233. +1 −1 nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/crash/CrashTest.java
  234. +4 −4 nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/crash/SpecialTests.java
  235. +1 −1 ...4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/dataset/MultiNormalizerMinMaxScalerTest.java
  236. +1 −1 ...d4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/dataset/MultiNormalizerStandardizeTest.java
  237. +3 −3 nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/dataset/NormalizerMinMaxScalerTest.java
  238. +7 −7 nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/dataset/NormalizerStandardizeTest.java
  239. +2 −2 nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/dataset/PreProcessor3D4DTest.java
  240. +20 −20 ...j-tests/src/test/java/org/nd4j/linalg/dataset/api/preprocessor/UnderSamplingPreProcessorTest.java
  241. +4 −4 nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/indexing/BooleanIndexingTest.java
  242. +2 −2 nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/lossfunctions/LossFunctionTest.java
  243. +8 −8 nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/mixed/MixedDataTypesTests.java
  244. +25 −38 nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/ops/DerivativeTests.java
  245. +18 −20 nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/ops/OpExecutionerTests.java
  246. +23 −23 nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/ops/OpExecutionerTestsC.java
  247. +1 −1 nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/ops/RationalTanhTest.java
  248. +1 −1 nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/rng/RandomTests.java
  249. +2 −2 nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/rng/RngValidationTests.java
  250. +5 −5 nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/shape/ShapeTestsC.java
  251. +1 −1 nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/specials/LongTests.java
  252. +1 −1 nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/util/ShapeTestC.java
  253. +1 −1 nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/linalg/workspace/BasicWorkspaceTests.java
  254. +1 −1 nd4j/nd4j-common/src/main/java/org/nd4j/linalg/util/ArrayUtil.java
@@ -88,7 +88,7 @@ public INDArray applyDropout(INDArray inputActivations, INDArray output, int ite
noise = workspaceMgr.createUninitialized(ArrayType.INPUT, inputActivations.shape(), inputActivations.ordering());
Nd4j.getExecutioner().exec(new GaussianDistribution(noise, 1.0, stdev));

return Nd4j.getExecutioner().execAndReturn(new OldMulOp(inputActivations, noise, output));
return Nd4j.getExecutioner().exec(new OldMulOp(inputActivations, noise, output));
}

@Override
@@ -96,8 +96,7 @@ public INDArray doForward(boolean training, LayerWorkspaceMgr workspaceMgr) {
case Subtract:
if (inputs.length != 2)
throw new IllegalArgumentException("ElementWise subtraction only supports 2 inputs");
return Nd4j.getExecutioner().execAndReturn(
new OldSubOp(inputs[0], inputs[1], workspaceMgr.createUninitialized(ArrayType.ACTIVATIONS, inputs[0].shape())));
return Nd4j.getExecutioner().exec(new OldSubOp(inputs[0], inputs[1], workspaceMgr.createUninitialized(ArrayType.ACTIVATIONS, inputs[0].shape())));
case Product:
INDArray product = workspaceMgr.dup(ArrayType.ACTIVATIONS, inputs[0]);
for (int i = 1; i < inputs.length; i++) {
@@ -86,7 +86,7 @@ public INDArray doForward(boolean training, LayerWorkspaceMgr workspaceMgr) {
return x.divColumnVector(xNorm2);
} else {
INDArray out = Nd4j.createUninitialized(x.shape(), x.ordering());
return Nd4j.getExecutioner().execAndReturn(new BroadcastDivOp(x, xNorm2, out, 0));
return Nd4j.getExecutioner().exec(new BroadcastDivOp(x, xNorm2, out, 0));
}
}
}
@@ -77,7 +77,7 @@ public INDArray doForward(boolean training, LayerWorkspaceMgr workspaceMgr) {
}

try(MemoryWorkspace ws = workspaceMgr.notifyScopeBorrowed(ArrayType.ACTIVATIONS)) {
return Nd4j.getExecutioner().exec(new EuclideanDistance(a, b), dimensions);
return Nd4j.getExecutioner().exec(new EuclideanDistance(a, b, dimensions));
}
}

@@ -112,7 +112,7 @@ public INDArray doForward(boolean training, LayerWorkspaceMgr workspaceMgr) {
}
} else {
//RNN and CNN case - Broadcast along dimension 0
dLda = Nd4j.getExecutioner().execAndReturn(new BroadcastMulOp(diff, first, diff, 0));
dLda = Nd4j.getExecutioner().exec(new BroadcastMulOp(diff, first, diff, 0));
try(MemoryWorkspace ws = workspaceMgr.notifyScopeBorrowed(ArrayType.ACTIVATION_GRAD)) {
dLdb = dLda.neg();
}
@@ -225,7 +225,7 @@ public Type type() {
.build();
Nd4j.getExecutioner().exec(op);

INDArray isMax = Nd4j.getExecutioner().execAndReturn(new IsMax(col2d, col2d, 1));
INDArray isMax = Nd4j.getExecutioner().exec(new IsMax(col2d, col2d, 1));
isMax.muliColumnVector(epsilon1d);
break;
case AVG:
@@ -279,7 +279,7 @@ These make zero difference for local training (other than perhaps when using FP1
dxhat = epsilon.mul(layerConf.getGamma());
} else {
//Standard case
dxhat = Nd4j.getExecutioner().execAndReturn(new BroadcastMulOp(epsilon, gamma,
dxhat = Nd4j.getExecutioner().exec(new BroadcastMulOp(epsilon, gamma,
Nd4j.createUninitialized(epsilon.shape(), epsilon.ordering()), 1));
}

@@ -292,9 +292,8 @@ These make zero difference for local training (other than perhaps when using FP1
INDArray dxmu2 = xMu.sum(0, 2, 3).muli(-2.0 / effectiveBatchSize).muli(dLdVar);
INDArray dLdmu = dxmu1.addi(dxmu2);

INDArray dLdx = Nd4j.getExecutioner().execAndReturn(new BroadcastDivOp(dxhat, std, dxhat, 1))
.addi(Nd4j.getExecutioner().execAndReturn(
new BroadcastMulOp(xMu, dLdVar.muli(2.0 / effectiveBatchSize), xMu, 1)));
INDArray dLdx = Nd4j.getExecutioner().exec(new BroadcastDivOp(dxhat, std, dxhat, 1))
.addi(Nd4j.getExecutioner().exec(new BroadcastMulOp(xMu, dLdVar.muli(2.0 / effectiveBatchSize), xMu, 1)));
Nd4j.getExecutioner()
.execAndReturn(new BroadcastAddOp(dLdx, dLdmu.muli(1.0 / effectiveBatchSize), dLdx, 1));

@@ -513,9 +512,9 @@ public INDArray preOutput(INDArray x, TrainingMode training, LayerWorkspaceMgr w
if (!Shape.strideDescendingCAscendingF(x))
x = x.dup(); //TODO: temp Workaround for broadcast bug. To be removed when fixed
xMu = workspaceMgr.createUninitialized(ArrayType.INPUT, x.shape(), x.ordering());
xMu = Nd4j.getExecutioner().execAndReturn(new BroadcastSubOp(x, mean,xMu, 1));
xMu = Nd4j.getExecutioner().exec(new BroadcastSubOp(x, mean,xMu, 1));
xHat = workspaceMgr.createUninitialized(ArrayType.INPUT, x.shape(), x.ordering());
xHat = Nd4j.getExecutioner().execAndReturn(new BroadcastDivOp(xMu, std,xHat, 1));
xHat = Nd4j.getExecutioner().exec(new BroadcastDivOp(xMu, std,xHat, 1));

if (layerConf.isLockGammaBeta()) {
//Special case: gamma/beta have fixed values for all outputs
@@ -531,10 +530,8 @@ public INDArray preOutput(INDArray x, TrainingMode training, LayerWorkspaceMgr w
} else {
//Standard case: gamma and beta are learned per parameter
activations = workspaceMgr.createUninitialized(ArrayType.ACTIVATIONS, x.shape(), x.ordering());
activations = Nd4j.getExecutioner().execAndReturn(
new BroadcastMulOp(xHat, gamma, activations, 1));
activations = Nd4j.getExecutioner()
.execAndReturn(new BroadcastAddOp(activations, beta, activations, 1));
activations = Nd4j.getExecutioner().exec(new BroadcastMulOp(xHat, gamma, activations, 1));
activations = Nd4j.getExecutioner().exec(new BroadcastAddOp(activations, beta, activations, 1));
}
} else {
// TODO setup BatchNorm for RNN http://arxiv.org/pdf/1510.01378v1.pdf
@@ -303,8 +303,8 @@ private INDArray epsilonHelperFullArray(INDArray inputArray, INDArray epsilon, i

switch (poolingType) {
case MAX:
INDArray isMax = Nd4j.getExecutioner().execAndReturn(new IsMax(inputArray.dup(), poolDim));
return Nd4j.getExecutioner().execAndReturn(new BroadcastMulOp(isMax, epsilon, isMax, broadcastDims));
INDArray isMax = Nd4j.getExecutioner().exec(new IsMax(inputArray.dup(), poolDim));
return Nd4j.getExecutioner().exec(new BroadcastMulOp(isMax, epsilon, isMax, broadcastDims));
case AVG:
//if out = avg(in,dims) then dL/dIn = 1/N * dL/dOut
int n = 1;
@@ -558,7 +558,7 @@ private static void cacheExit(boolean training, CacheMode cacheMode, LayerWorksp
INDArray deltao = deltaoNext;
Nd4j.getExecutioner().exec(new OldMulOp(nablaOut, sigmahOfS, deltao));
if (sigmoidGates) {
INDArray sigmaoPrimeOfZo = Nd4j.getExecutioner().execAndReturn(new TimesOneMinus(ao.dup('f'))); //Equivalent to sigmoid deriv on zo
INDArray sigmaoPrimeOfZo = Nd4j.getExecutioner().exec(new TimesOneMinus(ao.dup('f'))); //Equivalent to sigmoid deriv on zo
deltao.muli(sigmaoPrimeOfZo);
} else {
deltao.assign(gateActivationFn.backprop(fwdPass.oz[time], deltao).getFirst()); //Deltao needs to be modified in-place
@@ -610,8 +610,7 @@ private static void cacheExit(boolean training, CacheMode cacheMode, LayerWorksp
deltag.muli(ai);
deltag.muli(nablaCellState);
} else {
INDArray temp2 = Nd4j.getExecutioner().execAndReturn(
new OldMulOp(ai, nablaCellState, Nd4j.createUninitialized(ai.shape(), 'f')));
INDArray temp2 = Nd4j.getExecutioner().exec(new OldMulOp(ai, nablaCellState, Nd4j.createUninitialized(ai.shape(), 'f')));
deltag.assign(gateActivationFn.backprop(fwdPass.gz[time], temp2).getFirst());
//TODO activation functions with params; optimize (no assign)
}
@@ -620,8 +619,7 @@ private static void cacheExit(boolean training, CacheMode cacheMode, LayerWorksp
//Network input delta:
INDArray zi = fwdPass.iz[time];
INDArray deltai = deltaiNext;
temp = Nd4j.getExecutioner().execAndReturn(
new OldMulOp(ag, nablaCellState, Nd4j.createUninitialized(deltai.shape(), 'f')));
temp = Nd4j.getExecutioner().exec(new OldMulOp(ag, nablaCellState, Nd4j.createUninitialized(deltai.shape(), 'f')));
deltai.assign(afn.backprop(zi, temp).getFirst());
//TODO activation functions with params; also: optimize this (no assign)
//Shape: [m,n^L]
@@ -221,8 +221,7 @@ public double optimize(INDArray parameters, INDArray gradients, INDArray searchD

// check for convergence on delta x
if ((step < stepMin) || Nd4j.getExecutioner()
.execAndReturn(new Eps(parameters, candidateParameters,
Nd4j.createUninitialized(DataType.BOOL, candidateParameters.shape(), candidateParameters.ordering()),
.exec(new Eps(parameters, candidateParameters,Nd4j.createUninitialized(DataType.BOOL, candidateParameters.shape(), candidateParameters.ordering()),
candidateParameters.length())).castTo(DataType.FLOAT).sumNumber().longValue() == candidateParameters.length()) {
score = setScoreFor(parameters, workspaceMgr);
log.debug("EXITING BACKTRACK: Jump too small (stepMin = {}). Exiting and using original params. Score = {}",
@@ -134,9 +134,9 @@ public static INDArray maskedPoolingEpsilonTimeSeries(PoolingType poolingType, I
Nd4j.getExecutioner().exec(new BroadcastAddOp(input, negInfMask, withInf, 0, 2));
//At this point: all the masked out steps have value -inf, hence can't be the output of the MAX op

INDArray isMax = Nd4j.getExecutioner().execAndReturn(new IsMax(withInf, 2));
INDArray isMax = Nd4j.getExecutioner().exec(new IsMax(withInf, 2));

return Nd4j.getExecutioner().execAndReturn(new BroadcastMulOp(isMax, epsilon2d, isMax, 0, 1));
return Nd4j.getExecutioner().exec(new BroadcastMulOp(isMax, epsilon2d, isMax, 0, 1));
case AVG:
case SUM:
//if out = sum(in,dims) then dL/dIn = dL/dOut -> duplicate to each step and mask
@@ -290,9 +290,9 @@ public static INDArray maskedPoolingEpsilonCnn(PoolingType poolingType, INDArray
Nd4j.getExecutioner().exec(new BroadcastAddOp(input, negInfMask, withInf, dimensions));
//At this point: all the masked out steps have value -inf, hence can't be the output of the MAX op

INDArray isMax = Nd4j.getExecutioner().execAndReturn(new IsMax(withInf, 2, 3));
INDArray isMax = Nd4j.getExecutioner().exec(new IsMax(withInf, 2, 3));

return Nd4j.getExecutioner().execAndReturn(new BroadcastMulOp(isMax, epsilon2d, isMax, 0, 1));
return Nd4j.getExecutioner().exec(new BroadcastMulOp(isMax, epsilon2d, isMax, 0, 1));
case AVG:
case SUM:
//if out = sum(in,dims) then dL/dIn = dL/dOut -> duplicate to each step and mask
@@ -119,7 +119,8 @@ class ND4J_EXPORT NativeOps {
void *extraParams,
void *hZ, Nd4jLong *hZShapeInfo,
void *dZ, Nd4jLong *dZShapeInfo,
int *dimension, int dimensionLength);
void *hDimension, Nd4jLong *hDimensionShape,
void *dDimension, Nd4jLong *dDimensionShape);

/**
*
@@ -142,7 +143,8 @@ class ND4J_EXPORT NativeOps {
void *dY, Nd4jLong *dYShapeInfo,
void *hZ, Nd4jLong *hZShapeInfo,
void *dZ, Nd4jLong *dZShapeInfo,
int *dimension,int dimensionLength);
void *hDimension, Nd4jLong *hDimensionShape,
void *dDimension, Nd4jLong *dDimensionShape);


void execBroadcastBool(
@@ -154,7 +156,8 @@ class ND4J_EXPORT NativeOps {
void *dY, Nd4jLong *dYShapeInfo,
void *hZ, Nd4jLong *hZShapeInfo,
void *dZ, Nd4jLong *dZShapeInfo,
int *dimension,int dimensionLength);
void *hDimension, Nd4jLong *hDimensionShape,
void *dDimension, Nd4jLong *dDimensionShape);

/**
*
@@ -248,7 +251,8 @@ class ND4J_EXPORT NativeOps {
void *extraParams,
void *hZ, Nd4jLong *hZShapeInfo,
void *dZ, Nd4jLong *dZShapeInfo,
int *dimension,int dimensionLength);
void *hDimension, Nd4jLong *hDimensionShape,
void *dDimension, Nd4jLong *dDimensionShape);


void execReduceSame(Nd4jPointer *extraPointers,
@@ -258,7 +262,8 @@ class ND4J_EXPORT NativeOps {
void *extraParams,
void *hZ, Nd4jLong *hZShapeInfo,
void *dZ, Nd4jLong *dZShapeInfo,
int *dimension, int dimensionLength);
void *hDimension, Nd4jLong *hDimensionShape,
void *dDimension, Nd4jLong *dDimensionShape);


void execReduceBool(Nd4jPointer *extraPointers,
@@ -268,7 +273,8 @@ class ND4J_EXPORT NativeOps {
void *extraParams,
void *hZ, Nd4jLong *hZShapeInfo,
void *dZ, Nd4jLong *dZShapeInfo,
int *dimension, int dimensionLength);
void *hDimension, Nd4jLong *hDimensionShape,
void *dDimension, Nd4jLong *dDimensionShape);


void execReduceLong(Nd4jPointer *extraPointers,
@@ -278,7 +284,8 @@ class ND4J_EXPORT NativeOps {
void *extraParams,
void *hZ, Nd4jLong *hZShapeInfo,
void *dZ, Nd4jLong *dZShapeInfo,
int *dimension, int dimensionLength);
void *hDimension, Nd4jLong *hDimensionShape,
void *dDimension, Nd4jLong *dDimensionShape);

/**
*
@@ -343,8 +350,8 @@ class ND4J_EXPORT NativeOps {
void *dY, Nd4jLong *dYShapeInfo,
void *hZ, Nd4jLong *hZShapeInfo,
void *dZ, Nd4jLong *dZShapeInfo,
int *dimension,
int dimensionLength,
void *hDimension, Nd4jLong *hDimensionShape,
void *dDimension, Nd4jLong *dDimensionShape,
Nd4jLong *tadOnlyShapeInfo, Nd4jLong *tadOffsets,
Nd4jLong *yTadOnlyShapeInfo, Nd4jLong *yTadOffsets);

@@ -358,7 +365,8 @@ class ND4J_EXPORT NativeOps {
void *dY, Nd4jLong *dYShapeInfo,
void *hZ, Nd4jLong *hZShapeInfo,
void *dZ, Nd4jLong *dZShapeInfo,
int *dimension, int dimensionLength,
void *hDimension, Nd4jLong *hDimensionShape,
void *dDimension, Nd4jLong *dDimensionShape,
Nd4jLong *xTadShapeInfo, Nd4jLong *xOffsets,
Nd4jLong *yTadShapeInfo, Nd4jLong *yOffsets);

@@ -443,8 +451,8 @@ class ND4J_EXPORT NativeOps {
void *extraParams,
void *hZ, Nd4jLong *hZShapeInfo,
void *dZ, Nd4jLong *dZShapeInfo,
int *dimension,
int dimensionLength,
void *hDimension, Nd4jLong *hDimensionShape,
void *dDimension, Nd4jLong *dDimensionShape,
bool biasCorrected,
Nd4jLong *tadShapeInfo, Nd4jLong *tadOffsets);

@@ -520,8 +528,8 @@ class ND4J_EXPORT NativeOps {
void *hScalars, Nd4jLong *hScalarShapeInfo,
void *dScalars, Nd4jLong *dScalarShapeInfo,
void *extraParams,
int *dimension,
int dimensionLength,
void *hDimension, Nd4jLong *hDimensionShape,
void *dDimension, Nd4jLong *dDimensionShape,
Nd4jLong *tadShapeInfo, Nd4jLong *tadOffsets,
Nd4jLong *tadShapeInfoZ, Nd4jLong *tadOffsetsZ);

@@ -534,8 +542,8 @@ class ND4J_EXPORT NativeOps {
void *hScalars, Nd4jLong *hScalarShapeInfo,
void *dScalars, Nd4jLong *dScalarShapeInfo,
void *extraParams,
int *dimension,
int dimensionLength,
void *hDimension, Nd4jLong *hDimensionShape,
void *dDimension, Nd4jLong *dDimensionShape,
Nd4jLong *tadShapeInfo, Nd4jLong *tadOffsets,
Nd4jLong *tadShapeInfoZ, Nd4jLong *tadOffsetsZ);

0 comments on commit b69b83e

Please sign in to comment.
You can’t perform that action at this time.