Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for host_graph (0.09 sec)

  1. 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)
  2. 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)
Back to top