Searched defs:rhs_cols (Results 1 – 8 of 8) 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() local428 const int rhs_cols = extended_rhs_shape.Dims(4); in BatchMatMul() local
78 const int rhs_cols = extended_rhs_shape.Dims(4); in BatchMatMul() local136 const int rhs_cols = extended_rhs_shape.Dims(4); in BatchMatMul() local225 const int rhs_cols = extended_rhs_shape.Dims(4); in BatchMatMul() local
49 int64_t rhs_cols = n; in MatMul() local87 int64_t rhs_cols = n; in MatMul_Batch() local
42 int64_t rhs_cols = n; in MatMul() local
106 auto rhs_cols = rhs.dim_size(ndims_rhs - 1); in Compute() local
81 auto rhs_cols = rhs.dim_size(ndims_rhs - 1); in MKLContractOperands() local
524 void TestMaybeValidGemm(int lhs_rows, int lhs_cols, int rhs_rows, int rhs_cols, in TestMaybeValidGemm()
6320 int64_t rhs_cols = (spec.rcd == 0) ? spec.n : spec.k; in TEST_P() local6379 int64_t rhs_cols = (spec.rcd == 0) ? spec.n : (spec.k + k_increase); in TEST_P() local