| /external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/ |
| D | real_nvp.py | 202 shift, log_scale = self._shift_and_log_scale_fn(
205 if log_scale is not None:
206 y1 *= math_ops.exp(log_scale)
216 shift, log_scale = self._shift_and_log_scale_fn(
221 if log_scale is not None:
222 x1 *= math_ops.exp(-log_scale)
229 _, log_scale = self._shift_and_log_scale_fn(
231 if log_scale is None:
233 return -math_ops.reduce_sum(log_scale, axis=-1)
238 _, log_scale = self._shift_and_log_scale_fn(
[all …]
|
| D | masked_autoregressive.py | 250 shift, log_scale = self._shift_and_log_scale_fn(y)
253 if log_scale is not None:
254 next_y *= math_ops.exp(log_scale)
278 shift, log_scale = self._shift_and_log_scale_fn(y0)
280 if log_scale is not None:
281 y *= math_ops.exp(log_scale)
293 shift, log_scale = self._shift_and_log_scale_fn(y)
297 if log_scale is not None:
298 x *= math_ops.exp(-log_scale)
302 _, log_scale = self._shift_and_log_scale_fn(y)
[all …]
|
| /external/libaom/libaom/av1/encoder/x86/ |
| D | av1_highbd_quantize_avx2.c | 33 const int16_t *dequant_ptr, int log_scale, in init_qp() argument
36 if (log_scale) { in init_qp()
37 const __m128i round_scale = _mm_set1_epi16(1 << (15 - log_scale)); in init_qp()
49 const int16_t *iscan_ptr, int log_scale, in quantize() argument
59 q_lo = _mm256_srli_epi64(q_lo, 16 - log_scale); in quantize()
60 q_hi = _mm256_srli_epi64(q_hi, 16 - log_scale); in quantize()
63 const __m256i abs_s = _mm256_slli_epi32(abs_coeff, 1 + log_scale); in quantize()
68 dq = _mm256_srai_epi32(dq, log_scale); in quantize()
95 const int16_t *scan, const int16_t *iscan, int log_scale) { in av1_highbd_quantize_fp_avx2() argument
102 init_qp(round_ptr, quant_ptr, dequant_ptr, log_scale, qp); in av1_highbd_quantize_fp_avx2()
[all …]
|
| D | av1_quantize_avx2.c | 46 const int16_t *dequant_ptr, int log_scale, in init_qp() argument
52 if (log_scale > 0) { in init_qp()
53 const __m128i rnd = _mm_set1_epi16((int16_t)1 << (log_scale - 1)); in init_qp()
55 round = _mm_srai_epi16(round, log_scale); in init_qp()
61 if (log_scale == 1) { in init_qp()
62 qp[1] = _mm256_slli_epi16(qp[1], log_scale); in init_qp()
66 *thr = _mm256_srai_epi16(qp[2], 1 + log_scale); in init_qp()
69 static INLINE void update_qp(int log_scale, __m256i *thr, __m256i *qp) { in update_qp() argument
73 *thr = _mm256_srai_epi16(qp[2], 1 + log_scale); in update_qp()
149 const int log_scale = 0; in av1_quantize_fp_avx2() local
[all …]
|
| D | av1_highbd_quantize_sse4.c | 120 const int16_t *scan, const int16_t *iscan, int log_scale) { in av1_highbd_quantize_fp_sse4_1() argument
126 const int shift = 16 - log_scale; in av1_highbd_quantize_fp_sse4_1()
137 const int round1 = ROUND_POWER_OF_TWO(round_ptr[1], log_scale); in av1_highbd_quantize_fp_sse4_1()
138 const int round0 = ROUND_POWER_OF_TWO(round_ptr[0], log_scale); in av1_highbd_quantize_fp_sse4_1()
147 quantize_coeff_phase1(&coeff[0], qparam, shift, log_scale, qcoeff, dequant, in av1_highbd_quantize_fp_sse4_1()
155 quantize_coeff_phase2(qcoeff, dequant, &coeff_sign, qparam, shift, log_scale, in av1_highbd_quantize_fp_sse4_1()
160 quantize_coeff_phase1(&coeff[1], qparam, shift, log_scale, qcoeff, dequant, in av1_highbd_quantize_fp_sse4_1()
162 quantize_coeff_phase2(qcoeff, dequant, &coeff_sign, qparam, shift, log_scale, in av1_highbd_quantize_fp_sse4_1()
179 quantize_coeff_phase1(&coeff[0], qparam, shift, log_scale, qcoeff, dequant, in av1_highbd_quantize_fp_sse4_1()
182 log_scale, quanAddr, dquanAddr); in av1_highbd_quantize_fp_sse4_1()
[all …]
|
| /external/libaom/libaom/av1/encoder/ |
| D | av1_quantize.c | 42 const qm_val_t *iqm_ptr, int log_scale) { in quantize_fp_helper_c() argument
44 const int rounding[2] = { ROUND_POWER_OF_TWO(round_ptr[0], log_scale), in quantize_fp_helper_c()
45 ROUND_POWER_OF_TWO(round_ptr[1], log_scale) }; in quantize_fp_helper_c()
63 if ((abs_coeff << (1 + log_scale)) >= thresh) { in quantize_fp_helper_c()
66 tmp32 = (int)((abs_coeff * quant_ptr[rc != 0]) >> (16 - log_scale)); in quantize_fp_helper_c()
70 (tmp32 * dequant_ptr[rc != 0]) >> log_scale; in quantize_fp_helper_c()
91 (dequant_ptr[rc != 0] << (AOM_QM_BITS - (1 + log_scale)))) { in quantize_fp_helper_c()
95 (16 - log_scale + AOM_QM_BITS)); in quantize_fp_helper_c()
97 const tran_low_t abs_dqcoeff = (tmp32 * dequant) >> log_scale; in quantize_fp_helper_c()
113 const qm_val_t *iqm_ptr, int log_scale) { in highbd_quantize_fp_helper_c() argument
[all …]
|
| /external/libaom/libaom/aom_dsp/ |
| D | quantize.c | 22 const qm_val_t *iqm_ptr, const int log_scale) { in quantize_b_adaptive_helper_c() argument
23 const int zbins[2] = { ROUND_POWER_OF_TWO(zbin_ptr[0], log_scale), in quantize_b_adaptive_helper_c()
24 ROUND_POWER_OF_TWO(zbin_ptr[1], log_scale) }; in quantize_b_adaptive_helper_c()
64 clamp(abs_coeff + ROUND_POWER_OF_TWO(round_ptr[rc != 0], log_scale), in quantize_b_adaptive_helper_c()
69 (16 - log_scale + AOM_QM_BITS)); // quantization in quantize_b_adaptive_helper_c()
75 const tran_low_t abs_dqcoeff = (tmp32 * dequant) >> log_scale; in quantize_b_adaptive_helper_c()
114 const qm_val_t *iqm_ptr, const int log_scale) { in quantize_b_helper_c() argument
115 const int zbins[2] = { ROUND_POWER_OF_TWO(zbin_ptr[0], log_scale), in quantize_b_helper_c()
116 ROUND_POWER_OF_TWO(zbin_ptr[1], log_scale) }; in quantize_b_helper_c()
149 clamp(abs_coeff + ROUND_POWER_OF_TWO(round_ptr[rc != 0], log_scale), in quantize_b_helper_c()
[all …]
|
| D | quantize.h | 29 const qm_val_t *iqm_ptr, const int log_scale);
60 const qm_val_t *iqm_ptr, const int log_scale);
90 const qm_val_t *iqm_ptr, const int log_scale);
105 const qm_val_t *iqm_ptr, const int log_scale);
|
| /external/libaom/libaom/aom_dsp/x86/ |
| D | adaptive_quantize_sse2.c | 224 const int log_scale = 1; in aom_quantize_b_32x32_adaptive_sse2() local
227 const __m128i log_scale_vec = _mm_set1_epi16(log_scale); in aom_quantize_b_32x32_adaptive_sse2()
233 const int zbins[2] = { ROUND_POWER_OF_TWO(zbin_ptr[0], log_scale), in aom_quantize_b_32x32_adaptive_sse2()
234 ROUND_POWER_OF_TWO(zbin_ptr[1], log_scale) }; in aom_quantize_b_32x32_adaptive_sse2()
273 zbin = _mm_srli_epi16(zbin, log_scale); in aom_quantize_b_32x32_adaptive_sse2()
274 round = _mm_srli_epi16(round, log_scale); in aom_quantize_b_32x32_adaptive_sse2()
307 calculate_qcoeff_log_scale(&qcoeff0, round, quant, &shift, &log_scale); in aom_quantize_b_32x32_adaptive_sse2()
313 calculate_qcoeff_log_scale(&qcoeff1, round, quant, &shift, &log_scale); in aom_quantize_b_32x32_adaptive_sse2()
327 &log_scale); in aom_quantize_b_32x32_adaptive_sse2()
330 dqcoeff_ptr + 8, &log_scale); in aom_quantize_b_32x32_adaptive_sse2()
[all …]
|
| D | quantize_x86.h | 53 const int *log_scale) { in calculate_qcoeff_log_scale() argument
59 tmp = _mm_srli_epi16(tmp, (16 - *log_scale)); in calculate_qcoeff_log_scale()
61 tmp1 = _mm_slli_epi16(tmp1, *log_scale); in calculate_qcoeff_log_scale()
73 const int *log_scale) { in calculate_dqcoeff_and_store_log_scale() argument
86 dqcoeff32_0 = _mm_srli_epi32(dqcoeff32_0, *log_scale); in calculate_dqcoeff_and_store_log_scale()
87 dqcoeff32_1 = _mm_srli_epi32(dqcoeff32_1, *log_scale); in calculate_dqcoeff_and_store_log_scale()
|
| /external/tensorflow/tensorflow/lite/experimental/microfrontend/lib/ |
| D | BUILD | 62 ":log_scale",
70 name = "log_scale",
73 "log_scale.c",
78 "log_scale.h",
156 ":log_scale",
|
| D | frontend_util.c | 27 LogScaleFillConfigWithDefaults(&config->log_scale); in FrontendFillConfigWithDefaults()
70 if (!LogScalePopulateState(&config->log_scale, &state->log_scale)) { in FrontendPopulateState()
|
| D | frontend_test.cc | 48 config_.log_scale.enable_log = true; in FrontendTestConfig()
49 config_.log_scale.scale_shift = 6; in FrontendTestConfig()
|
| D | frontend_util.h | 35 struct LogScaleConfig log_scale; member
|
| D | frontend.h | 38 struct LogScaleState log_scale; member
|
| /external/libaom/libaom/test/ |
| D | av1_quantize_test.cc | 30 const int16_t *scan, const int16_t *iscan, int log_scale);
69 int log_scale = (txSize == TX_32X32); in RunQuantizeTest() local
98 &ref_eob, scanOrder.scan, scanOrder.iscan, log_scale); in RunQuantizeTest()
103 scanOrder.scan, scanOrder.iscan, log_scale)); in RunQuantizeTest()
142 int log_scale = (txSize == TX_32X32); in RunEobTest() local
175 &ref_eob, scanOrder.scan, scanOrder.iscan, log_scale); in RunEobTest()
180 scanOrder.scan, scanOrder.iscan, log_scale)); in RunEobTest()
|
| D | quantize_func_test.cc | 38 typedef void (*QuantizeFuncHbd)(QUAN_PARAM_LIST, int log_scale);
42 qcoeff_ptr, dqcoeff_ptr, dequant_ptr, eob_ptr, scan, iscan, log_scale)
50 const int log_scale = 0; in highbd_quan16x16_wrapper() local
56 const int log_scale = 1; in highbd_quan32x32_wrapper() local
62 const int log_scale = 2; in highbd_quan64x64_wrapper() local
|
| /external/tensorflow/tensorflow/lite/experimental/micro/examples/micro_speech/micro_features/ |
| D | frontend_util.cc | 26 LogScaleFillConfigWithDefaults(&config->log_scale); in FrontendFillConfigWithDefaults()
70 if (!LogScalePopulateState(error_reporter, &config->log_scale, in FrontendPopulateState()
71 &state->log_scale)) { in FrontendPopulateState()
|
| D | BUILD | 109 ":log_scale",
120 name = "log_scale",
123 "log_scale.cc",
128 "log_scale.h",
254 ":log_scale",
|
| D | micro_features_generator.cc | 48 config.log_scale.enable_log = 1; in InitializeMicroFeatures()
49 config.log_scale.scale_shift = 6; in InitializeMicroFeatures()
|
| D | frontend_test.cc | 48 config_.log_scale.enable_log = true; in FrontendTestConfig()
49 config_.log_scale.scale_shift = 6; in FrontendTestConfig()
|
| D | frontend_util.h | 33 struct LogScaleConfig log_scale; member
|
| D | frontend.h | 34 struct LogScaleState log_scale; member
|
| /external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/bijectors/ |
| D | real_nvp_test.py | 139 log_scale = constant_op.constant([0.5])
140 return shift, log_scale
|
| /external/tensorflow/tensorflow/lite/experimental/microfrontend/ |
| D | audio_microfrontend.cc | 64 config.log_scale.enable_log = m["enable_log"].AsBool(); in Init()
65 config.log_scale.scale_shift = m["scale_shift"].AsInt32(); in Init()
|