MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / TEST_P

Function TEST_P

tensorflow/compiler/xla/service/cpu/cpu_runtime_test.cc:150–161  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

148};
149
150TEST_P(EigenMatMulTest, DoIt) {
151 MatMulShape shape = std::get<0>(GetParam());
152 bool transpose_lhs = std::get<1>(GetParam());
153 bool transpose_rhs = std::get<2>(GetParam());
154 bool single_threaded = std::get<3>(GetParam());
155
156 auto a = MakeLinspaceArray2D(0.0, 1.0, shape.m, shape.k);
157 auto b = MakeLinspaceArray2D(-2.0, 2.0, shape.k, shape.n);
158 auto c = EigenMatrixMultiply(*a, *b, transpose_lhs, transpose_rhs,
159 single_threaded);
160 CheckMatrixMultiply(*a, *b, *c);
161}
162
163INSTANTIATE_TEST_SUITE_P(EigenMatMulTestInstantiaion, EigenMatMulTest,
164 ::testing::Combine(::testing::ValuesIn(MatMulShapes),

Callers

nothing calls this directly

Calls 5

GetParamFunction · 0.85
MakeLinspaceArray2DFunction · 0.85
EigenMatrixMultiplyFunction · 0.85
CheckMatrixMultiplyFunction · 0.85
MKLMatrixMultiplyFunction · 0.85

Tested by

no test coverage detected