Searched defs:candidate_weight_shape (Results 1 – 2 of 2) sorted by relevance
36 const RuntimeShape& candidate_weight_shape, in GruCell()
51 const RuntimeShape candidate_weight_shape = GetTensorShape(candidate_weight); in GruImpl() local