Searched refs:rhs_cols (Results 1 – 7 of 7) sorted by relevance
77 const int rhs_cols = extended_rhs_shape.Dims(4); in BatchMatMul() local88 rhs_params.cols = rhs_cols; in BatchMatMul()93 dst_params.cols = rhs_cols; in BatchMatMul()106 lhs_rows * rhs_cols; in BatchMatMul()168 const int rhs_cols = extended_rhs_shape.Dims(4); in BatchMatMul() local171 const int ioff_ext0 = rhs_ext0 == 0 ? 0 : rhs_cols; in BatchMatMul()172 const int ioff_ext1 = rhs_ext1 == 0 ? 0 : rhs_cols; in BatchMatMul()173 const int ioff_ext2 = rhs_ext2 == 0 ? 0 : rhs_cols; in BatchMatMul()198 rhs_params.cols = rhs_cols; in BatchMatMul()203 dst_params.cols = rhs_cols; in BatchMatMul()[all …]
77 const int rhs_cols = extended_rhs_shape.Dims(4); in BatchMatMul() local91 lhs_rows * rhs_cols; in BatchMatMul()92 for (int j = 0; j < rhs_cols; ++j) { in BatchMatMul()135 const int rhs_cols = extended_rhs_shape.Dims(4); in BatchMatMul() local138 const int ioff_ext0 = rhs_ext0 == 0 ? 0 : rhs_cols; in BatchMatMul()139 const int ioff_ext1 = rhs_ext1 == 0 ? 0 : rhs_cols; in BatchMatMul()140 const int ioff_ext2 = rhs_ext2 == 0 ? 0 : rhs_cols; in BatchMatMul()177 lhs_rows * rhs_cols; in BatchMatMul()178 for (int j = 0; j < rhs_cols; ++j) { in BatchMatMul()224 const int rhs_cols = extended_rhs_shape.Dims(4); in BatchMatMul() local[all …]
43 tensorflow::int64 rhs_cols = n; in MatMul() local45 std::swap(rhs_rows, rhs_cols); in MatMul()51 rhs_cols); in MatMul()
50 tensorflow::int64 rhs_cols = n; in MatMul() local52 std::swap(rhs_rows, rhs_cols); in MatMul()58 rhs_cols); in MatMul()
109 auto rhs_cols = rhs.dim_size(ndims_rhs - 1); in Compute() local112 if (adj_y_) std::swap(rhs_rows, rhs_cols); in Compute()120 out_shape.AddDim(rhs_cols); in Compute()
780 int rhs_cols = end[Side::kRhs] - start[Side::kRhs]; \802 rhs_cols / kernel_rhs_cols); \
5623 int64 rhs_cols = (spec.rcd == 0) ? spec.n : spec.k; in TEST_P() local5624 Shape rhs_shape = ShapeUtil::MakeShape(F32, {rhs_rows, rhs_cols}); in TEST_P()5628 /*cols=*/rhs_cols))); in TEST_P()5682 int64 rhs_cols = (spec.rcd == 0) ? spec.n : (spec.k + k_increase); in TEST_P() local5683 Shape rhs_shape = ShapeUtil::MakeShape(F32, {rhs_rows, rhs_cols}); in TEST_P()5687 /*cols=*/rhs_cols))); in TEST_P()