Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for edges (0.26 sec)

  1. tensorflow/c/c_api.cc

      mutex_lock l(graph->mu);
      std::vector<const tensorflow::Edge*> control_edges;
      for (const tensorflow::Edge* edge : op->node.in_edges()) {
        if (!edge->IsControlEdge()) continue;
        control_edges.push_back(edge);
      }
      for (const tensorflow::Edge* edge : control_edges) {
        graph->graph.RemoveControlEdge(edge);
      }
    }
    
    void TF_SetRequireShapeInferenceFns(TF_Graph* graph, bool require) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 102.3K bytes
    - Viewed (0)
  2. tensorflow/c/c_api_function_test.cc

        }
    
        // If caller specified all edges, check that we have seen all
        if (is_exact_edges) {
          ASSERT_EQ(e_edges.size() + c_edges.size(), a_edges.size())
              << "Expected edges: " << ToString(e_edges)
              << " Expected Control edges: " << ToString(c_edges)
              << " Actual edges: " << ToString(a_edges)
              << " in fdef: " << fdef.DebugString();
        }
      }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jul 20 22:08:54 GMT 2023
    - 63.6K bytes
    - Viewed (6)
  3. tensorflow/c/c_api_test.cc

      ASSERT_EQ(1, TF_OperationNumInputs(neg));
      TF_Output neg_input = TF_OperationInput({neg, 0});
      EXPECT_EQ(scalar, neg_input.oper);
      EXPECT_EQ(0, neg_input.index);
    
      // Test that we can't see control edges involving the source and sink nodes.
      TF_Operation* control_ops[100];
      EXPECT_EQ(0, TF_OperationNumControlInputs(scalar));
      EXPECT_EQ(0, TF_OperationGetControlInputs(scalar, control_ops, 100));
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 96.9K bytes
    - Viewed (3)
Back to top