Commit e35a9afe authored by Oleg Dzhimiev's avatar Oleg Dzhimiev
Browse files

working on GeForce GTX 1050 Ti

parent 763c5251
Loading
Loading
Loading
Loading
+0 −27
Original line number Diff line number Diff line
@@ -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;
}