Searched refs:rank_0 (Results 1 – 1 of 1) sorted by relevance
66 Output rank_0 = get_stacked_rank(inputs->at(0)); in ExpandDimsForBroadcast() local69 Output max_rank = ops::Maximum(scope, rank_0, rank_1); in ExpandDimsForBroadcast()108 *inputs = VectorizerInput({expand_dims_if_unstacked(inputs->at(0), rank_0), in ExpandDimsForBroadcast()