Searched defs:rank_diff (Results 1 – 2 of 2) sorted by relevance
79 Output rank_diff = ops::Sub(scope, max_rank, rank); in ExpandDimsForBroadcast() local
865 int rank_diff = shape_x.size() - shape_y.size(); in AreBroadcastCompatible() local1342 auto rank_diff = output_type.getRank() - input_type.getRank(); in matchAndRewrite() local