Searched refs:rank_y (Results 1 – 2 of 2) sorted by relevance
2002 const int32 rank_y = c->Rank(shape_y); in BroadcastBinaryOpOutputShapeFnHelper() local2003 const int32 rank_out = std::max(rank_x, rank_y); in BroadcastBinaryOpOutputShapeFnHelper()2010 if (rank_x != rank_y) dim_one = c->MakeDim(1); in BroadcastBinaryOpOutputShapeFnHelper()2015 const bool dim_y_is_one = (i < (rank_out - rank_y)); in BroadcastBinaryOpOutputShapeFnHelper()2017 dim_y_is_one ? dim_one : c->Dim(shape_y, i - (rank_out - rank_y)); in BroadcastBinaryOpOutputShapeFnHelper()
92 int rank_y = input_shape_y.dimensions_count(); in ComputeBinaryOperatorOutputSize() local93 int rank_out = std::max(rank_x, rank_y); 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()