Searched defs:rhs_cols (Results 1 – 6 of 6) 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
77 const int rhs_cols = extended_rhs_shape.Dims(4); in BatchMatMul() local135 const int rhs_cols = extended_rhs_shape.Dims(4); in BatchMatMul() local224 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
43 tensorflow::int64 rhs_cols = n; in MatMul() local
50 tensorflow::int64 rhs_cols = n; in MatMul() local
5623 int64 rhs_cols = (spec.rcd == 0) ? spec.n : spec.k; in TEST_P() local5682 int64 rhs_cols = (spec.rcd == 0) ? spec.n : (spec.k + k_increase); in TEST_P() local