Searched refs:rank_1 (Results 1 – 2 of 2) sorted by relevance
67 Output rank_1 = get_stacked_rank(inputs->at(1)); in ExpandDimsForBroadcast() local69 Output max_rank = ops::Maximum(scope, rank_0, rank_1); in ExpandDimsForBroadcast()109 expand_dims_if_unstacked(inputs->at(1), rank_1)}); in ExpandDimsForBroadcast()
1769 rank_1 = array_ops.expand_dims(array_ops.rank(images) - 1, 0)1770 shape_list = ([array_ops.ones(rank_1, dtype=dtypes.int32)] +