Searched refs:rhs_strides (Results 1 – 3 of 3) sorted by relevance
204 auto rhs_strides = CalculateTFStrides(rhs_dims); in CreateMatMulParams() local223 rhs_strides[k_idx] = k; in CreateMatMulParams()224 rhs_strides[n_idx] = 1; in CreateMatMulParams()227 lhs_dims, rhs_dims, out_dims, lhs_strides, rhs_strides, out_strides); in CreateMatMulParams()
216 array<Index, RDims> rhs_strides;217 rhs_strides[0] = 1;219 rhs_strides[i+1] = rhs_strides[i] * eval_right_dims[i];283 m_right_nocontract_strides[nocontract_idx] = rhs_strides[i];310 m_right_contracting_strides[i] = rhs_strides[right];
252 %rhs_strides = alloca %ArrayType, align 8