Searched refs:dim_y_is_one (Results 1 – 2 of 2) sorted by relevance
101 bool dim_y_is_one = i < (rank_out - rank_y); in ComputeBinaryOperatorOutputSize() local102 int dim_y = dim_y_is_one ? 1 : input_shape_y.dims(i - (rank_out - rank_y)); in ComputeBinaryOperatorOutputSize()108 if (dim_x == 1 && !dim_y_is_one) { in ComputeBinaryOperatorOutputSize()
2106 const bool dim_y_is_one = (i < (rank_out - rank_y)); in BroadcastBinaryOpOutputShapeFnHelper() local2108 dim_y_is_one ? dim_one : c->Dim(shape_y, i - (rank_out - rank_y)); in BroadcastBinaryOpOutputShapeFnHelper()2147 if (c->Value(dim_x) == 1 && !dim_y_is_one) { in BroadcastBinaryOpOutputShapeFnHelper()