Searched refs:candidate_bias_shape (Results 1 – 3 of 3) sorted by relevance
38 const RuntimeShape& candidate_bias_shape, in GruCell() argument81 candidate_weight, candidate_bias_shape, candidate_bias, in GruCell()
33 const RuntimeShape& candidate_bias_shape,
53 const RuntimeShape candidate_bias_shape = GetTensorShape(candidate_bias); in GruImpl() local69 candidate_weight_shape, candidate_weight_data, candidate_bias_shape, in GruImpl()