mirror of
https://gitlab.com/libeigen/eigen.git
synced 2024-12-21 07:19:46 +08:00
39baff850c
TernaryFunctors and their executors allow operations on 3-tuples of inputs. API fully implemented for Arrays and Tensors based on binary functors. Ported the cephes betainc function (regularized incomplete beta integral) to Eigen, with support for CPU and GPU, floats, doubles, and half types. Added unit tests in array.cpp and cxx11_tensor_cuda.cu Collapsed revision * Merged helper methods for betainc across floats and doubles. * Added TensorGlobalFunctions with betainc(). Removed betainc() from TensorBase. * Clean up CwiseTernaryOp checks, change igamma_helper to cephes_helper. * betainc: merge incbcf and incbd into incbeta_cfe. and more cleanup. * Update TernaryOp and SpecialFunctions (betainc) based on review comments.
134 lines
3.8 KiB
Plaintext
134 lines
3.8 KiB
Plaintext
// This file is part of Eigen, a lightweight C++ template library
|
|
// for linear algebra.
|
|
//
|
|
// Copyright (C) 2014 Benoit Steiner <benoit.steiner.goog@gmail.com>
|
|
// Copyright (C) 2013 Christian Seiler <christian@iwakd.de>
|
|
//
|
|
// This Source Code Form is subject to the terms of the Mozilla
|
|
// Public License v. 2.0. If a copy of the MPL was not distributed
|
|
// with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
//#ifndef EIGEN_CXX11_TENSOR_MODULE
|
|
//#define EIGEN_CXX11_TENSOR_MODULE
|
|
|
|
#include "../../../Eigen/Core"
|
|
|
|
#include <Eigen/src/Core/util/DisableStupidWarnings.h>
|
|
|
|
#include "src/util/CXX11Meta.h"
|
|
#include "src/util/MaxSizeVector.h"
|
|
|
|
/** \defgroup CXX11_Tensor_Module Tensor Module
|
|
*
|
|
* This module provides a Tensor class for storing arbitrarily indexed
|
|
* objects.
|
|
*
|
|
* \code
|
|
* #include <Eigen/CXX11/Tensor>
|
|
* \endcode
|
|
*/
|
|
|
|
#include <cmath>
|
|
#include <cstddef>
|
|
#include <cstring>
|
|
|
|
#ifdef _WIN32
|
|
typedef __int32 int32_t;
|
|
typedef unsigned __int32 uint32_t;
|
|
typedef __int64 int64_t;
|
|
typedef unsigned __int64 uint64_t;
|
|
#else
|
|
#include <stdint.h>
|
|
#endif
|
|
|
|
#if __cplusplus > 199711 || EIGEN_COMP_MSVC >= 1900
|
|
#include <random>
|
|
#endif
|
|
|
|
#ifdef _WIN32
|
|
#include <windows.h>
|
|
#elif defined(__APPLE__)
|
|
#include <mach/mach_time.h>
|
|
#else
|
|
#include <time.h>
|
|
#endif
|
|
|
|
#ifdef EIGEN_USE_THREADS
|
|
#include "ThreadPool"
|
|
#endif
|
|
|
|
#ifdef EIGEN_USE_GPU
|
|
#include <iostream>
|
|
#include <cuda_runtime.h>
|
|
#if defined(__CUDACC__)
|
|
#include <curand_kernel.h>
|
|
#endif
|
|
#endif
|
|
|
|
#include "src/Tensor/TensorMacros.h"
|
|
#include "src/Tensor/TensorForwardDeclarations.h"
|
|
#include "src/Tensor/TensorMeta.h"
|
|
#include "src/Tensor/TensorFunctors.h"
|
|
#include "src/Tensor/TensorCostModel.h"
|
|
#include "src/Tensor/TensorDeviceDefault.h"
|
|
#include "src/Tensor/TensorDeviceThreadPool.h"
|
|
#include "src/Tensor/TensorDeviceCuda.h"
|
|
#include "src/Tensor/TensorIndexList.h"
|
|
#include "src/Tensor/TensorDimensionList.h"
|
|
#include "src/Tensor/TensorDimensions.h"
|
|
#include "src/Tensor/TensorInitializer.h"
|
|
#include "src/Tensor/TensorTraits.h"
|
|
#include "src/Tensor/TensorUInt128.h"
|
|
#include "src/Tensor/TensorIntDiv.h"
|
|
#include "src/Tensor/TensorGlobalFunctions.h"
|
|
|
|
#include "src/Tensor/TensorBase.h"
|
|
|
|
#include "src/Tensor/TensorEvaluator.h"
|
|
#include "src/Tensor/TensorExpr.h"
|
|
#include "src/Tensor/TensorReduction.h"
|
|
#include "src/Tensor/TensorReductionCuda.h"
|
|
#include "src/Tensor/TensorArgMax.h"
|
|
#include "src/Tensor/TensorConcatenation.h"
|
|
#include "src/Tensor/TensorContractionMapper.h"
|
|
#include "src/Tensor/TensorContractionBlocking.h"
|
|
#include "src/Tensor/TensorContraction.h"
|
|
#include "src/Tensor/TensorContractionThreadPool.h"
|
|
#include "src/Tensor/TensorContractionCuda.h"
|
|
#include "src/Tensor/TensorConversion.h"
|
|
#include "src/Tensor/TensorConvolution.h"
|
|
#include "src/Tensor/TensorFFT.h"
|
|
#include "src/Tensor/TensorPatch.h"
|
|
#include "src/Tensor/TensorImagePatch.h"
|
|
#include "src/Tensor/TensorVolumePatch.h"
|
|
#include "src/Tensor/TensorBroadcasting.h"
|
|
#include "src/Tensor/TensorChipping.h"
|
|
#include "src/Tensor/TensorInflation.h"
|
|
#include "src/Tensor/TensorLayoutSwap.h"
|
|
#include "src/Tensor/TensorMorphing.h"
|
|
#include "src/Tensor/TensorPadding.h"
|
|
#include "src/Tensor/TensorReverse.h"
|
|
#include "src/Tensor/TensorShuffling.h"
|
|
#include "src/Tensor/TensorStriding.h"
|
|
#include "src/Tensor/TensorCustomOp.h"
|
|
#include "src/Tensor/TensorEvalTo.h"
|
|
#include "src/Tensor/TensorForcedEval.h"
|
|
#include "src/Tensor/TensorGenerator.h"
|
|
#include "src/Tensor/TensorAssign.h"
|
|
#include "src/Tensor/TensorScan.h"
|
|
|
|
#include "src/Tensor/TensorExecutor.h"
|
|
#include "src/Tensor/TensorDevice.h"
|
|
|
|
#include "src/Tensor/TensorStorage.h"
|
|
#include "src/Tensor/Tensor.h"
|
|
#include "src/Tensor/TensorFixedSize.h"
|
|
#include "src/Tensor/TensorMap.h"
|
|
#include "src/Tensor/TensorRef.h"
|
|
|
|
#include "src/Tensor/TensorIO.h"
|
|
|
|
#include <Eigen/src/Core/util/ReenableStupidWarnings.h>
|
|
|
|
//#endif // EIGEN_CXX11_TENSOR_MODULE
|