Searched refs:dim_y_is_one (Results 1 – 2 of 2) sorted by relevance
1335 const bool dim_y_is_one = (i < (rank_out - rank_y)); in BroadcastBinaryOpOutputShapeFnHelper() local1337 dim_y_is_one ? dim_one : c->Dim(shape_y, i - (rank_out - rank_y)); in BroadcastBinaryOpOutputShapeFnHelper()1361 if (c->Value(dim_x) == 1 && !dim_y_is_one) { in BroadcastBinaryOpOutputShapeFnHelper()
100 bool dim_y_is_one = i < (rank_out - rank_y); in ComputeBinaryOperatorOutputSize() local101 int dim_y = dim_y_is_one ? 1 : input_shape_y.dims(i - (rank_out - rank_y)); in ComputeBinaryOperatorOutputSize()107 if (dim_x == 1 && !dim_y_is_one) { in ComputeBinaryOperatorOutputSize()