Searched refs:lhs_rank (Results 1 – 6 of 6) sorted by relevance
80 auto lhs_rank = lhs_ranked_type.getRank(); in CreateBroadcastsForBinaryOp() local86 GetI64ElementsAttrForSeq(0, std::max(lhs_rank, rhs_rank), rewriter); in CreateBroadcastsForBinaryOp()89 if (lhs_rank > rhs_rank) { in CreateBroadcastsForBinaryOp()92 } else if (lhs_rank < rhs_rank) { in CreateBroadcastsForBinaryOp()
44 const int64 lhs_rank = lhs_shape.rank(); in CanonicalizeDot() local46 lhs_rank - num_batch_dims - num_contracting_dims; in CanonicalizeDot()54 for (int64 i = 0; i < lhs_rank; ++i) { in CanonicalizeDot()68 lhs_transpose.reserve(lhs_rank); in CanonicalizeDot()
1045 int64 lhs_rank = lhs->shape().rank(); in ComputeArrayForDotWithIndexedLhs() local1048 0, lhs->source_dim() == (lhs_rank - 1) ? (lhs_rank - 2) : (lhs_rank - 1)); in ComputeArrayForDotWithIndexedLhs()
1059 const auto lhs_rank = lhs_shape.rank();1062 CHECK_EQ(num_spatial_dims + 2, lhs_rank);1251 const int64 lhs_rank = lhs->shape().rank();1269 if (lhs_rank == 2 && rhs_rank == 2 && lhs_contracting_dimension == 1 &&1312 const auto lhs_rank = lhs->shape().rank();1324 DimensionVector lhs_index(lhs_rank);1332 (lhs_rank - dnums.lhs_contracting_dimensions_size()) +1344 for (int64 i = 0; i < lhs_rank; i++) {
258 const int32 lhs_rank = c->Rank(lhs_shape_handle); in __anon00ae58f50202() local259 for (int64 i = 0; i < lhs_rank; ++i) { in __anon00ae58f50202()
570 const int64 lhs_rank = lhs_shape->rank(); in BinaryOp() local576 if (!broadcast_dimensions.empty() && lhs_rank != rhs_rank) { in BinaryOp()577 const bool should_broadcast_lhs = lhs_rank < rhs_rank; in BinaryOp()