Searched defs:lhs_rows (Results 1 – 6 of 6) 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
76 const int lhs_rows = extended_lhs_shape.Dims(3); in BatchMatMul() local134 const int lhs_rows = extended_lhs_shape.Dims(3); in BatchMatMul() local223 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
36 tensorflow::int64 lhs_rows = m; in MatMul() local
43 tensorflow::int64 lhs_rows = m; in MatMul() local
5600 int64 lhs_rows = (spec.lcd == 0) ? (spec.k + k_increase) : spec.m; in TEST_P() local5670 int64 lhs_rows = (spec.lcd == 0) ? spec.k : spec.m; in TEST_P() local