Searched refs:get_stacked_rank (Results 1 – 1 of 1) sorted by relevance
55 auto get_stacked_rank = [&scope](const WrappedTensor& input) { in ExpandDimsForBroadcast() local66 Output rank_0 = get_stacked_rank(inputs->at(0)); in ExpandDimsForBroadcast()67 Output rank_1 = get_stacked_rank(inputs->at(1)); in ExpandDimsForBroadcast()