Searched refs:dim_one (Results 1 – 1 of 1) sorted by relevance
1329 DimensionHandle dim_one; in BroadcastBinaryOpOutputShapeFnHelper() local1330 if (rank_x != rank_y) dim_one = c->MakeDim(1); in BroadcastBinaryOpOutputShapeFnHelper()1333 ? dim_one in BroadcastBinaryOpOutputShapeFnHelper()1337 dim_y_is_one ? dim_one : c->Dim(shape_y, i - (rank_out - rank_y)); in BroadcastBinaryOpOutputShapeFnHelper()