- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for host_graph (0.09 sec)
-
tensorflow/compiler/jit/extract_outside_compilation_pass_test.cc
host_func_attrs["_device_ordinal"] = device_ordinal_temp_value; TF_CHECK_OK(FunctionDefToBodyHelper( *fld.Find("host_graph"), AttrSlice(&host_func_attrs), &fld, &host_fbody)); Graph *host_graph = host_fbody->graph; Node *key_placeholder = nullptr, *sequencer = nullptr; for (Node *n : host_graph->nodes()) { if (n->type_string() == "Placeholder" && absl::EndsWith(n->name(), "_key_placeholder")) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 41K bytes - Viewed (0) -
tensorflow/cc/training/queue_runner_test.cc
// Second call to run dequeue op is to make sure the cost graph has been // stored. TF_EXPECT_OK(session->Run({}, {kDequeueOp0}, {}, &dq0)); CostGraphDef cost_graph; TF_CHECK_OK(qr->ExportCostGraph(&cost_graph)); EXPECT_TRUE(cost_graph.node_size() > 0); qr->Stop(session.get()); } TEST(QueueRunnerTest, NoRunMetaDataTest) { GraphDef graph_def = BuildSimpleGraph();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Sep 21 06:27:51 UTC 2019 - 14.7K bytes - Viewed (0)