Searched refs:rank_0 (Results 1 – 2 of 2) 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()
757 %rank_0 = rank %const_0 : tensor<2x1x4xi32>760 return %rank_0 : index769 %rank_0 = rank %arg0 : memref<?x?xf32>772 return %rank_0 : index