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