onnxruntime/test/providers/cpu/math/gemm_test.cc
// Disable TensorRT on some of the tests because TensorRT only support FLOAT, INT8, FLOAT16 and INT32 for now
Disable TensorRT on some of the tests because TensorRT only support FLOAT, INT8, FLOAT16 and INT32 for now
Disable TensorRT on some of the tests because TensorRT only support FLOAT INT8 FLOAT16 and INT32 for now
TEST(GemmOpTest, GemmNoTrans) {
  OpTester test("Gemm");
  test.AddAttribute("transA", (int64_t)0);
  test.AddAttribute("transB", (int64_t)0);
  test.AddAttribute("alpha", 1.0f);
  test.AddAttribute("beta", 1.0f);
  test.AddInput<float>("A", {2, 4},
                       {1.0f, 2.0f, 3.0f, 4.0f,
                        -1.0f, -2.0f, -3.0f, -4.0f});
  test.AddInput<float>("B", {4, 3}, std::vector<float>(12, 1.0f));
  test.AddInput<float>("C", {2, 3}, std::vector<float>(6, 1.0f));
  test.AddOutput<float>("Y", {2, 3},
                        {11.0f, 11.0f, 11.0f,
                         -9.0f, -9.0f, -9.0f});
  test.Run(OpTester::ExpectResult::kExpectSuccess, "", {kTensorrtExecutionProvider});
}
TEST ( GemmOpTest , GemmNoTrans ) { OpTester test ( " " ) ; test . AddAttribute ( " " , ( int64_t ) 0 ) ; test . AddAttribute ( " " , ( int64_t ) 0 ) ; test . AddAttribute ( " " , 1.0f ) ; test . AddAttribute ( " " , 1.0f ) ; test . AddInput < float > ( " " , { 2 , 4 } , { 1.0f , 2.0f , 3.0f , 4.0f , -1.0f , -2.0f , -3.0f , -4.0f } ) ; test . AddInput < float > ( " " , { 4 , 3 } , std :: vector < float > ( 12 , 1.0f ) ) ; test . AddInput < float > ( " " , { 2 , 3 } , std :: vector < float > ( 6 , 1.0f ) ) ; test . AddOutput < float > ( " " , { 2 , 3 } , { 11.0f , 11.0f , 11.0f , -9.0f , -9.0f , -9.0f } ) ; test . Run ( OpTester :: ExpectResult :: kExpectSuccess , " " , { kTensorrtExecutionProvider } ) ; }
Disable TensorRT on some of the tests because TensorRT only support FLOAT, INT8, FLOAT16 and INT32 for now
Disable TensorRT on some of the tests because TensorRT only support FLOAT INT8 FLOAT16 and INT32 for now