Searched refs:input_shape_y (Results 1 – 1 of 1) sorted by relevance
84 const Shape& input_shape_y, in ComputeBinaryOperatorOutputSize() argument91 int rank_y = input_shape_y.dimensions_count(); in ComputeBinaryOperatorOutputSize()101 int dim_y = dim_y_is_one ? 1 : input_shape_y.dims(i - (rank_out - rank_y)); in ComputeBinaryOperatorOutputSize()