Home
last modified time | relevance | path

Searched refs:scale_v (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/distributions/
Dlaplace_test.py61 scale_v = 3.0
68 expected_log_pdf = stats.laplace.logpdf(x, loc_v, scale=scale_v)
80 scale_v = np.array([3.0, 4.0])
92 expected_log_pdf = stats.laplace.logpdf(x, loc_v, scale=scale_v)
101 scale_v = 3.0
113 expected_log_pdf = stats.laplace.logpdf(x, loc_v, scale=scale_v)
122 scale_v = 3.0
131 expected_cdf = stats.laplace.cdf(x, loc_v, scale=scale_v)
139 scale_v = 3.0
148 expected_cdf = stats.laplace.logcdf(x, loc_v, scale=scale_v)
[all …]
/external/webp/sharpyuv/
Dsharpyuv_csp.c34 float scale_v = cb; in SharpYuvComputeConversionMatrix() local
41 scale_v *= (224 << shift) / denom; in SharpYuvComputeConversionMatrix()
55 matrix->rgb_to_v[0] = ToFixed16((1 - kr) * scale_v); in SharpYuvComputeConversionMatrix()
56 matrix->rgb_to_v[1] = ToFixed16(-kg * scale_v); in SharpYuvComputeConversionMatrix()
57 matrix->rgb_to_v[2] = ToFixed16(-kb * scale_v); in SharpYuvComputeConversionMatrix()
/external/tensorflow/tensorflow/core/kernels/
Dfused_batch_norm_op.cu.cc149 U scale_v = scale[channel]; in run() local
154 U scaling_factor_v = rsqrt(var_v + epsilon) * scale_v; in run()
229 half2 scale_v = __floats2half2_rn(scale[channel[0]], scale[channel[1]]); in run() local
236 __hmul2(h2rsqrt(__hadd2(var_v, epsilon_h2)), scale_v); in run()
265 half scale_v = __float2half(scale[channel]); in run() local
270 half scaling_factor_v = __hmul(hrsqrt(__hadd(var_v, epsilon_h)), scale_v); in run()
/external/ComputeLibrary/src/cpu/kernels/pool2d/neon/nchw/
Dall.cpp129 const float16x4_t scale_v = vdup_n_f16(scale); in pooling3_fp16_neon_nchw() local
133 res = vmul_f16(vpadd_f16(res, res), scale_v); in pooling3_fp16_neon_nchw()
293 const float16x4_t scale_v = vdup_n_f16(scale); in pooling2_fp16_neon_nchw() local
296 res = vmul_f16(vpadd_f16(sum_data, sum_data), scale_v); in pooling2_fp16_neon_nchw()
543 const float32x2_t scale_v = vdup_n_f32(scale); in pooling2_fp32_neon_nchw() local
547 res = vmul_f32(vpadd_f32(sum_data, sum_data), scale_v); in pooling2_fp32_neon_nchw()
623 const float32x2_t scale_v = vdup_n_f32(scale); in pooling3_fp32_neon_nchw() local
628 res = vmul_f32(vpadd_f32(res, res), scale_v); in pooling3_fp32_neon_nchw()
692 const float32x2_t scale_v = vdup_n_f32(scale); in pooling7_fp32_neon_nchw() local
718 res = vmul_f32(vpadd_f32(res, res), scale_v); in pooling7_fp32_neon_nchw()
/external/ComputeLibrary/src/cpu/kernels/pool3d/neon/
Dquantized.h169 const float32x4_t scale_v = vdupq_n_f32(scale); in avg_poolingMxNxD_q8_neon_ndhwc()
171 … vres1 = vcvtq_q32_f32<q32x4_t>(wrapper::vmla(half_scale_v, vcvtq_f32_q32(vres1), scale_v)); in avg_poolingMxNxD_q8_neon_ndhwc()
172 … vres2 = vcvtq_q32_f32<q32x4_t>(wrapper::vmla(half_scale_v, vcvtq_f32_q32(vres2), scale_v)); in avg_poolingMxNxD_q8_neon_ndhwc()
173 … vres3 = vcvtq_q32_f32<q32x4_t>(wrapper::vmla(half_scale_v, vcvtq_f32_q32(vres3), scale_v)); in avg_poolingMxNxD_q8_neon_ndhwc()
174 … vres4 = vcvtq_q32_f32<q32x4_t>(wrapper::vmla(half_scale_v, vcvtq_f32_q32(vres4), scale_v)); in avg_poolingMxNxD_q8_neon_ndhwc()
Dimpl.cpp211 const vector_type scale_v = wrapper::vdup_n(static_cast<T>(scale), tag_type()); in avg_poolingMxNxD_fp_neon_ndhwc() local
235 vres = wrapper::vmul(vres, scale_v); in avg_poolingMxNxD_fp_neon_ndhwc()
360 const vector_type scale_v = wrapper::vdup_n(static_cast<T>(scale), tag_type()); in l2_poolingMxNxD_fp_neon_ndhwc() local
363 vres = wrapper::vmul(vres, scale_v); in l2_poolingMxNxD_fp_neon_ndhwc()
/external/ComputeLibrary/src/cpu/kernels/pool2d/neon/
Dquantized.h150 const float32x4_t scale_v = vdupq_n_f32(scale); in poolingMxN_q8_neon_nhwc()
152 … vres1 = vcvtq_q32_f32<q32x4_t>(wrapper::vmla(half_scale_v, vcvtq_f32_q32(vres1), scale_v)); in poolingMxN_q8_neon_nhwc()
153 … vres2 = vcvtq_q32_f32<q32x4_t>(wrapper::vmla(half_scale_v, vcvtq_f32_q32(vres2), scale_v)); in poolingMxN_q8_neon_nhwc()
154 … vres3 = vcvtq_q32_f32<q32x4_t>(wrapper::vmla(half_scale_v, vcvtq_f32_q32(vres3), scale_v)); in poolingMxN_q8_neon_nhwc()
155 … vres4 = vcvtq_q32_f32<q32x4_t>(wrapper::vmla(half_scale_v, vcvtq_f32_q32(vres4), scale_v)); in poolingMxN_q8_neon_nhwc()
Dfp32.cpp198 const float32x4_t scale_v = vdupq_n_f32(scale); in poolingMxN_fp32_neon_nhwc() local
222 vres = vmulq_f32(vres, scale_v); in poolingMxN_fp32_neon_nhwc()
Dfp16.cpp204 const float16x8_t scale_v = vdupq_n_f16(scale); in poolingMxN_fp16_neon_nhwc() local
227 vres = vmulq_f16(vres, scale_v); in poolingMxN_fp16_neon_nhwc()