Searched refs:diag_value (Results 1 – 2 of 2) sorted by relevance
464 diag_value = self.trace() / math_ops.cast(diag_size, self.dtype)468 return diag_value[..., array_ops.newaxis] * ones_diag
89 const int diag_value = ((i == 2 || i == 5) ? (1 << GM_ALPHA_PREC_BITS) : 0); in convert_to_params() local92 (int32_t)clamp(model[i] - diag_value, GM_ALPHA_MIN, GM_ALPHA_MAX); in convert_to_params()94 model[i] = (model[i] + diag_value) * GM_ALPHA_DECODE_FACTOR; in convert_to_params()