Searched refs:rhs_rank (Results 1 – 1 of 1) sorted by relevance
910 const auto rhs_rank = ShapeUtil::Rank(rhs_shape); in HandleConvolution() local913 CHECK_EQ(num_spatial_dims + 2, rhs_rank); in HandleConvolution()947 DimensionVector rhs_index(rhs_rank); in HandleConvolution()1036 const auto rhs_rank = ShapeUtil::Rank(rhs->shape()); in HandleDot() local1075 for (int64 i = 0; i < rhs_rank; i++) { in HandleDot()1085 DimensionVector rhs_index(rhs_rank); in HandleDot()