Searched refs:sqrt_ (Results 1 – 3 of 3) sorted by relevance
83 explicit SqrtGradientFunction(AbstractTensorHandle* sqrt) : sqrt_(sqrt) { in SqrtGradientFunction()90 TF_RETURN_IF_ERROR(SqrtGrad(ctx, {sqrt_.get(), grad_outputs[0]}, in Compute()97 AbstractTensorHandlePtr sqrt_; member in tensorflow::gradients::__anonb85f18960111::SqrtGradientFunction
140 SI F sqrt_(F v) { return sqrtf(v); } in sqrt_() function205 SI F sqrt_(F v) { return vsqrtq_f32(v); }214 SI F sqrt_(F v) {316 SI F sqrt_(F v) { return _mm256_sqrt_ps (v); }520 SI F sqrt_(F v) { return _mm_sqrt_ps (v); }2008 r = sqrt_(X2 + Y2); in STAGE()2017 t = x + sqrt_(ctx->fP0 - y*y); // ctx->fP0 = r0 * r0 in STAGE()2027 t = sqrt_(x*x + y*y) - x * ctx->fP0; // ctx->fP0 = 1/r1 in STAGE()2032 t = sqrt_(x*x - y*y) - x * ctx->fP0; // ctx->fP0 = 1/r1 in STAGE()2037 t = -sqrt_(x*x - y*y) - x * ctx->fP0; // ctx->fP0 = 1/r1 in STAGE()[all …]
125 SI F sqrt_(F v) { return sqrtf(v); } in sqrt_() function206 SI F sqrt_(F v) { return vsqrtq_f32(v); }215 SI F sqrt_(F v) {359 SI F sqrt_(F v) { return _mm256_sqrt_ps (v); }689 SI F sqrt_(F v) { return _mm_sqrt_ps (v); }1452 liteDst = sqrt_(m) - m, in BLEND_MODE()2488 r = sqrt_(X2 + Y2); in STAGE()2497 t = x + sqrt_(ctx->fP0 - y*y); // ctx->fP0 = r0 * r0 in STAGE()2507 t = sqrt_(x*x + y*y) - x * ctx->fP0; // ctx->fP0 = 1/r1 in STAGE()2512 t = sqrt_(x*x - y*y) - x * ctx->fP0; // ctx->fP0 = 1/r1 in STAGE()[all …]