Searched refs:rank_out (Results 1 – 2 of 2) sorted by relevance
93 int rank_out = std::max(rank_x, rank_y); in ComputeBinaryOperatorOutputSize() local96 dims_out->reserve(rank_out); in ComputeBinaryOperatorOutputSize()97 for (int i = 0; i < rank_out; ++i) { in ComputeBinaryOperatorOutputSize()98 int dim_x = i < (rank_out - rank_x) in ComputeBinaryOperatorOutputSize()100 : input_shape_x.dims(i - (rank_out - rank_x)); in ComputeBinaryOperatorOutputSize()101 bool dim_y_is_one = i < (rank_out - rank_y); in ComputeBinaryOperatorOutputSize()102 int dim_y = dim_y_is_one ? 1 : input_shape_y.dims(i - (rank_out - rank_y)); in ComputeBinaryOperatorOutputSize()
2094 const int32_t rank_out = std::max(rank_x, rank_y); in BroadcastBinaryOpOutputShapeFnHelper() local2102 for (int i = 0; i < rank_out; ++i) { in BroadcastBinaryOpOutputShapeFnHelper()2103 const auto dim_x = i < (rank_out - rank_x) in BroadcastBinaryOpOutputShapeFnHelper()2105 : c->Dim(shape_x, i - (rank_out - rank_x)); in BroadcastBinaryOpOutputShapeFnHelper()2106 const bool dim_y_is_one = (i < (rank_out - rank_y)); in BroadcastBinaryOpOutputShapeFnHelper()2108 dim_y_is_one ? dim_one : c->Dim(shape_y, i - (rank_out - rank_y)); in BroadcastBinaryOpOutputShapeFnHelper()