Пример #1
0
//vdPowx		Raising each vector element to the constant power
void klVSLPowx(klVector<double>& v,double x,klVector<double>& ans)
{
	const __int64_t n = v.getColumns();


	vdPowx( n,  v.getMemory(),x,ans.getMemory());

}
Пример #2
0
 /*
  * Class:     com_intel_analytics_bigdl_mkl_MKL
  * Method:    vdPowx
  * Signature: (I[DID[DI)V
  */
JNIEXPORT void JNICALL Java_com_intel_analytics_bigdl_mkl_MKL_vdPowx
   (JNIEnv * env, jclass cls, jint n, jdoubleArray a, jint aOffset, jdouble b, jdoubleArray y,
   jint yOffset) {

   jdouble * jni_a = (*env)->GetPrimitiveArrayCritical(env, a, JNI_FALSE);
   jdouble * jni_y = (*env)->GetPrimitiveArrayCritical(env, y, JNI_FALSE);

   vdPowx( n, jni_a + aOffset, b, jni_y + yOffset);

   (*env)->ReleasePrimitiveArrayCritical(env, y, jni_y, 0);
   (*env)->ReleasePrimitiveArrayCritical(env, a, jni_a, 0);
 }
Пример #3
0
void caffe_powx<double>(const int n, const double* a, const double b,
                        double* y) {
    vdPowx(n, a, b, y);
}