Searched refs:sub_k (Results 1 – 1 of 1) sorted by relevance
1456 int64 sub_k = concat_op->shape().dimensions(lhs_contracting_dim); in OptimizeDotOfConcatHelper() local1458 rhs_slice_shape.set_dimensions(rhs_contracting_dim, sub_k); in OptimizeDotOfConcatHelper()1465 limit_indices[rhs_contracting_dim] = rhs_contracting_dim_offset + sub_k; in OptimizeDotOfConcatHelper()1498 rhs_contracting_dim_offset += sub_k; in OptimizeDotOfConcatHelper()