Beispiel #1
0
            static SafeImportGraphDefResultsHandle ImportGraph(Status s, Graph graph, Buffer graph_def, Operation scalar)
            {
                using var opts = c_api.TF_NewImportGraphDefOptions();
                c_api.TF_ImportGraphDefOptionsSetPrefix(opts, "imported2");
                c_api.TF_ImportGraphDefOptionsAddInputMapping(opts, "scalar", 0, new TF_Output(scalar, 0));
                c_api.TF_ImportGraphDefOptionsAddReturnOutput(opts, "feed", 0);
                c_api.TF_ImportGraphDefOptionsAddReturnOutput(opts, "scalar", 0);
                EXPECT_EQ(2, c_api.TF_ImportGraphDefOptionsNumReturnOutputs(opts));
                c_api.TF_ImportGraphDefOptionsAddReturnOperation(opts, "scalar");
                EXPECT_EQ(1, c_api.TF_ImportGraphDefOptionsNumReturnOperations(opts));
                var results = c_api.TF_GraphImportGraphDefWithResults(graph, graph_def.Handle, opts, s.Handle);

                EXPECT_EQ(TF_Code.TF_OK, s.Code);

                return(results);
            }