Searched defs:rhs_cols (Results 1 – 8 of 8) sorted by relevance
77 const int rhs_cols = extended_rhs_shape.Dims(4); in BatchMatMul() local168 const int rhs_cols = extended_rhs_shape.Dims(4); in BatchMatMul() local322 const int rhs_cols = extended_rhs_shape.Dims(4); in BatchMatMul() local
78 const int rhs_cols = extended_rhs_shape.Dims(4); in BatchMatMul() local136 const int rhs_cols = extended_rhs_shape.Dims(4); in BatchMatMul() local225 const int rhs_cols = extended_rhs_shape.Dims(4); in BatchMatMul() local
109 auto rhs_cols = rhs.dim_size(ndims_rhs - 1); in Compute() local
80 auto rhs_cols = rhs.dim_size(ndims_rhs - 1); in MKLContractOperands() local
43 int64_t rhs_cols = n; in MatMul() local
50 int64_t rhs_cols = n; in MatMul() local
509 void TestMaybeValidGemm(int lhs_rows, int lhs_cols, int rhs_rows, int rhs_cols, in TestMaybeValidGemm()
5712 int64_t rhs_cols = (spec.rcd == 0) ? spec.n : spec.k; in TEST_P() local5771 int64_t rhs_cols = (spec.rcd == 0) ? spec.n : (spec.k + k_increase); in TEST_P() local