Home
last modified time | relevance | path

Searched refs:fanins (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/utils/
Dgraph_view.cc478 std::vector<std::vector<TensorId>> fanins; in MutableGraphView() local
479 Status s = CheckFaninsInternal(&fanins); in MutableGraphView()
485 AddFaninsInternal(&fanins); in MutableGraphView()
503 std::vector<std::vector<TensorId>>* fanins) { in CheckFaninsInternal() argument
505 fanins->reserve(num_nodes); in CheckFaninsInternal()
535 fanins->push_back(std::move(node_fanins)); in CheckFaninsInternal()
541 std::vector<std::vector<TensorId>>* fanins) { in AddFaninsInternal() argument
546 std::vector<TensorId>& node_fanins = fanins->at(i); in AddFaninsInternal()
Dgraph_view.h421 Status CheckFaninsInternal(std::vector<std::vector<TensorId>>* fanins);
423 void AddFaninsInternal(std::vector<std::vector<TensorId>>* fanins);
/external/tensorflow/tensorflow/core/grappler/
Dgraph_view_test.cc172 absl::flat_hash_set<string> fanins; in TEST_F() local
175 fanins.insert(absl::StrCat(fi.node->name(), ":", fi.port_id)); in TEST_F()
178 EXPECT_EQ(fanins, expected_fanins); in TEST_F()
Dmutable_graph_view_test.cc37 absl::Span<const string> fanins) { in CompareNodeFanins() argument
38 ASSERT_EQ(node->input_size(), fanins.size()); in CompareNodeFanins()
40 TensorId tensor_id = ParseTensorName(fanins[i]); in CompareNodeFanins()
72 absl::Span<const string> fanins, in CheckNode() argument
84 CompareNodeFanins(graph, node, fanins); in CheckNode()
Dmutable_graph_view.cc342 absl::flat_hash_set<absl::string_view> fanins; in AddAndDedupFanouts() local
357 if (!gtl::InsertIfNotPresent(&fanins, input_node_name) && in AddAndDedupFanouts()