Searched defs:lhs_rows (Results 1 – 8 of 8) sorted by relevance
76 const int lhs_rows = extended_lhs_shape.Dims(3); in BatchMatMul() local167 const int lhs_rows = extended_lhs_shape.Dims(3); in BatchMatMul() local321 const int lhs_rows = extended_lhs_shape.Dims(3); in BatchMatMul() local
77 const int lhs_rows = extended_lhs_shape.Dims(3); in BatchMatMul() local135 const int lhs_rows = extended_lhs_shape.Dims(3); in BatchMatMul() local224 const int lhs_rows = extended_lhs_shape.Dims(3); in BatchMatMul() local
106 auto lhs_rows = lhs.dim_size(ndims_lhs - 2); in Compute() local
77 auto lhs_rows = lhs.dim_size(ndims_lhs - 2); in MKLContractOperands() local
36 int64_t lhs_rows = m; in MatMul() local
43 int64_t lhs_rows = m; in MatMul() local
509 void TestMaybeValidGemm(int lhs_rows, int lhs_cols, int rhs_rows, int rhs_cols, in TestMaybeValidGemm()
5689 int64_t lhs_rows = (spec.lcd == 0) ? (spec.k + k_increase) : spec.m; in TEST_P() local5759 int64_t lhs_rows = (spec.lcd == 0) ? spec.k : spec.m; in TEST_P() local