Loading main.cpp +0 −27 Original line number Diff line number Diff line Loading @@ -194,32 +194,5 @@ int main(int, char**) { session->ReleaseCallable(feed_gpu_fetch_cpu); /* using namespace tensorflow; using namespace tensorflow::ops; auto root = Scope::NewRootScope(); //auto A = Const(root, {1,16,16,1},{8,8}); //auto B = Const(root, 5,{}); auto A = Placeholder(root.WithOpName("A"), DT_INT32, Placeholder::Shape({2,2})); auto v = Identity(root.WithOpName("v"), A); //auto v = Fill(root.WithOpName("v"), A, B); // Vector b = [3 5] //auto b = Const(root, {{3, 5}}); // v = Ab^T //auto v = MatMul(root.WithOpName("v"), A, b, MatMul::TransposeB(true)); ClientSession session2(root); // Run and fetch v int aa[2][2] = {{1,2},{3,4}}; TF_CHECK_OK(session2.Run({ {A,{{1,2},{3,4}}} },{v}, &outputs)); // Expect outputs[0] == [19; -3] LOG(INFO) << outputs[0].DebugString(); cout << outputs[0].matrix<int>() << endl; */ return 0; } Loading
main.cpp +0 −27 Original line number Diff line number Diff line Loading @@ -194,32 +194,5 @@ int main(int, char**) { session->ReleaseCallable(feed_gpu_fetch_cpu); /* using namespace tensorflow; using namespace tensorflow::ops; auto root = Scope::NewRootScope(); //auto A = Const(root, {1,16,16,1},{8,8}); //auto B = Const(root, 5,{}); auto A = Placeholder(root.WithOpName("A"), DT_INT32, Placeholder::Shape({2,2})); auto v = Identity(root.WithOpName("v"), A); //auto v = Fill(root.WithOpName("v"), A, B); // Vector b = [3 5] //auto b = Const(root, {{3, 5}}); // v = Ab^T //auto v = MatMul(root.WithOpName("v"), A, b, MatMul::TransposeB(true)); ClientSession session2(root); // Run and fetch v int aa[2][2] = {{1,2},{3,4}}; TF_CHECK_OK(session2.Run({ {A,{{1,2},{3,4}}} },{v}, &outputs)); // Expect outputs[0] == [19; -3] LOG(INFO) << outputs[0].DebugString(); cout << outputs[0].matrix<int>() << endl; */ return 0; }