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() local427 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
42 int64_t lhs_rows = m; in MatMul() local80 int64_t lhs_rows = m; in MatMul_Batch() local
35 int64_t lhs_rows = m; in MatMul() local
103 auto lhs_rows = lhs.dim_size(ndims_lhs - 2); in Compute() local
78 auto lhs_rows = lhs.dim_size(ndims_lhs - 2); in MKLContractOperands() local
524 void TestMaybeValidGemm(int lhs_rows, int lhs_cols, int rhs_rows, int rhs_cols, in TestMaybeValidGemm()
6297 int64_t lhs_rows = (spec.lcd == 0) ? (spec.k + k_increase) : spec.m; in TEST_P() local6367 int64_t lhs_rows = (spec.lcd == 0) ? spec.k : spec.m; in TEST_P() local