Commit e35a9afe authored by Oleg Dzhimiev's avatar Oleg Dzhimiev

working on GeForce GTX 1050 Ti

parent 763c5251
......@@ -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;
}
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment