Searched refs:lhs_contracting_dimension (Results 1 – 3 of 3) sorted by relevance
1230 const int64 lhs_contracting_dimension = dnums.lhs_contracting_dimensions(0); in HandleDot() local1233 CHECK_EQ(lhs->shape().dimensions(lhs_contracting_dimension), in HandleDot()1236 << lhs->shape().dimensions(lhs_contracting_dimension) in HandleDot()1241 if (lhs_rank == 2 && rhs_rank == 2 && lhs_contracting_dimension == 1 && in HandleDot()1252 lhs->shape().dimensions(lhs_contracting_dimension); in HandleDot()
646 const int64 lhs_contracting_dimension = in InferDotOpShape() local650 if (lhs.dimensions(lhs_contracting_dimension) != in InferDotOpShape()652 lhs.is_dynamic_dimension(lhs_contracting_dimension) != in InferDotOpShape()
1550 int lhs_contracting_dimension = dnums.lhs_contracting_dimensions(0); in OptimizeDotOfGather() local1554 lhs, rhs, /*a_contracting_dimension=*/lhs_contracting_dimension) && in OptimizeDotOfGather()1589 const int m = left_operand->shape().dimensions(1 - lhs_contracting_dimension); in OptimizeDotOfGather()1600 ? 1 - lhs_contracting_dimension in OptimizeDotOfGather()