Searched refs:candidate_weight_shape (Results 1 – 3 of 3) sorted by relevance
36 const RuntimeShape& candidate_weight_shape, in GruCell() argument80 FullyConnected(fc_params, concat_shape, concat, candidate_weight_shape, in GruCell()
31 const RuntimeShape& candidate_weight_shape,
51 const RuntimeShape candidate_weight_shape = GetTensorShape(candidate_weight); in GruImpl() local69 candidate_weight_shape, candidate_weight_data, candidate_bias_shape, in GruImpl()