wehub/deeplearning4j--deeplearning4j
1
0
Fork 0
Code Issues Pull Requests Actions Deployments Models Agent Notes Packages Projects Releases Wiki Activity

chore: import upstream snapshot with attribution

Browse Source
This commit is contained in:
wehub-resource-sync
2026-07-13 12:47:05 +08:00
commit 4f3b7da785
7394 changed files with 2005594 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libnd4j/include/legacy/cpu/NativeOpExecutioner_summary.cpp
+58
View File
@@ -0,0 +1,58 @@
/* ******************************************************************************
*
* SummaryStats operations - uses SD_COMMON_TYPES and SD_FLOAT_TYPES
*
******************************************************************************/
// Selective rendering - MUST be included before types.h to define HAS_* flags
#include <system/selective_rendering/core.h>
#include <system/selective_rendering/bool_types.h>
#include <system/selective_rendering/float_types.h>
#include <system/selective_rendering/bfloat_types.h>
#include <system/selective_rendering/int_types.h>
#include <system/selective_rendering/uint_types.h>
#include <legacy/NativeOpExecutioner.h>
#include <loops/summarystatsreduce.h>
#include <types/types.h>
////////////////////////////////////////////////////////////////////////
void NativeOpExecutioner::execSummaryStats(sd::LaunchContext *lc, int opNum, const void *hX,
sd::LongType *hXShapeInfo, const void *dX,
sd::LongType *dXShapeInfo, void *extraParams, void *hZ,
sd::LongType *hZShapeInfo, void *dZ, sd::LongType *dZShapeInfo,
bool biasCorrected) {
auto xType = sd::ArrayOptions::dataType(hXShapeInfo);
auto zType = sd::ArrayOptions::dataType(hZShapeInfo);
BUILD_DOUBLE_SELECTOR(xType, zType, functions::summarystats::SummaryStatsReduce,
::exec(opNum, biasCorrected, const_cast<void *>(hX), hXShapeInfo, extraParams, hZ, hZShapeInfo, nullptr, 1),
SD_COMMON_TYPES, SD_FLOAT_TYPES);
}
////////////////////////////////////////////////////////////////////////
void NativeOpExecutioner::execSummaryStatsScalar(sd::LaunchContext *lc, int opNum, const void *hX,
sd::LongType *hXShapeInfo, const void *dX,
sd::LongType *dXShapeInfo, void *extraParams, void *hZ,
sd::LongType *hZShapeInfo, void *dZ,
sd::LongType *dZShapeInfo, bool biasCorrected) {
auto xType = sd::ArrayOptions::dataType(hXShapeInfo);
auto zType = sd::ArrayOptions::dataType(hZShapeInfo);
BUILD_DOUBLE_SELECTOR(xType, zType, functions::summarystats::SummaryStatsReduce,
::execScalar(opNum, biasCorrected, const_cast<void *>(hX), hXShapeInfo, extraParams, hZ, hZShapeInfo),
SD_COMMON_TYPES, SD_FLOAT_TYPES);
}
////////////////////////////////////////////////////////////////////////
void NativeOpExecutioner::execSummaryStats(sd::LaunchContext *lc, int opNum, const void *hX,
sd::LongType *hXShapeInfo, const void *dX,
sd::LongType *dXShapeInfo, void *extraParams, void *hZ,
sd::LongType *hZShapeInfo, void *dZ, sd::LongType *dZShapeInfo,
sd::LongType *dimension, sd::LongType dimensionLength, sd::LongType *tadShapeInfo,
sd::LongType *tadOffsets, bool biasCorrected) {
auto xType = sd::ArrayOptions::dataType(hXShapeInfo);
auto zType = sd::ArrayOptions::dataType(hZShapeInfo);
BUILD_DOUBLE_SELECTOR(
xType, zType, functions::summarystats::SummaryStatsReduce,
::exec(opNum, biasCorrected, const_cast<void *>(hX), hXShapeInfo, extraParams, hZ, hZShapeInfo, dimension, dimensionLength),
SD_COMMON_TYPES, SD_FLOAT_TYPES);
}