Searched defs:lhs_cols (Results 1 – 6 of 6) sorted by relevance
43 int64_t lhs_cols = k; in MatMul() local81 int64_t lhs_cols = k; in MatMul_Batch() local
36 int64_t lhs_cols = k; in MatMul() local
104 auto lhs_cols = lhs.dim_size(ndims_lhs - 1); in Compute() local
79 auto lhs_cols = lhs.dim_size(ndims_lhs - 1); in MKLContractOperands() local
524 void TestMaybeValidGemm(int lhs_rows, int lhs_cols, int rhs_rows, int rhs_cols, in TestMaybeValidGemm()
6298 int64_t lhs_cols = (spec.lcd == 0) ? spec.m : (spec.k + k_increase); in TEST_P() local6368 int64_t lhs_cols = (spec.lcd == 0) ? spec.m : spec.k; in TEST_P() local