libidx
|
00001 /*************************************************************************** 00002 * Copyright (C) 2012 by Soumith Chintala * 00003 * soumith@gmail.com * 00004 * All rights reserved. 00005 * 00006 * Redistribution and use in source and binary forms, with or without 00007 * modification, are permitted provided that the following conditions are met: 00008 * * Redistributions of source code must retain the above copyright 00009 * notice, this list of conditions and the following disclaimer. 00010 * * Redistributions in binary form must reproduce the above copyright 00011 * notice, this list of conditions and the following disclaimer in the 00012 * documentation and/or other materials provided with the distribution. 00013 * * Redistribution under a license not approved by the Open Source 00014 * Initiative (http://www.opensource.org) must display the 00015 * following acknowledgement in all advertising material: 00016 * This product includes software developed at the Courant 00017 * Institute of Mathematical Sciences (http://cims.nyu.edu). 00018 * * The names of the authors may not be used to endorse or promote products 00019 * derived from this software without specific prior written permission. 00020 * 00021 * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESS OR IMPLIED 00022 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED 00023 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 00024 * DISCLAIMED. IN NO EVENT SHALL ThE AUTHORS BE LIABLE FOR ANY 00025 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES 00026 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 00027 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND 00028 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 00029 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 00030 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 00031 ***************************************************************************/ 00032 00033 #ifndef TH_H 00034 #define TH_H 00035 00036 #include "config.h" 00037 #include "numerics.h" 00038 #include "idx.h" 00039 #include "thops.h" 00040 00041 #ifdef __TH__ 00042 00043 namespace ebl { 00044 00045 // idx_add (not-in-place) //////////////////////////////////////////////////// 00046 00048 template<> void idx_add(idx<float32> &i1, idx<float32> &i2, 00049 idx<float32> &out); 00051 template<> void idx_add(idx<float64> &i1, idx<float64> &i2, 00052 idx<float64> &out); 00053 00054 // idx_add (in-place) //////////////////////////////////////////////////////// 00055 00057 template<> void idx_add(idx<float32> &in, idx<float32> &out); 00059 template<> void idx_add(idx<float64> &in, idx<float64> &out); 00060 00061 // idx_copy ////////////////////////////////////////////////////////////////// 00062 00064 template <> void idx_copy(const idx<float32> &src, idx<float32> &dst); 00066 template <> void idx_copy(const idx<float64> &src, idx<float64> &dst); 00067 00068 // idx_tanh //////////////////////////////////////////////////////////////// 00070 template <> void idx_tanh(idx<float32> &inp, idx<float32> &out); 00071 template <> void idx_tanh(idx<float64> &inp, idx<float64> &out); 00072 /* //! derivative of hyperbolic tangent */ 00073 /* template <> void idx_dtanh(idx<float32> &inp, idx<float32> &out); */ 00074 /* template <> void idx_dtanh(idx<float64> &inp, idx<float64> &out); */ 00075 00076 00077 } // end namespace ebl 00078 00079 #include "idxops_th.hpp" 00080 00081 #endif /* ifdef __TH__ */ 00082 00083 #endif /* define TH_H */