Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/utils/
Dpattern_utils.cc32 node_view->NumControlledFanouts() > 0) in DoesOpTypePatternMatch()
Dgraph_view_test.cc631 TYPED_TEST(TypedNodeViewTest, NumControlledFanouts) { in TYPED_TEST() argument
647 EXPECT_EQ(a_node->NumControlledFanouts(), 0); in TYPED_TEST()
648 EXPECT_EQ(b_node->NumControlledFanouts(), 0); in TYPED_TEST()
649 EXPECT_EQ(c_node->NumControlledFanouts(), 2); in TYPED_TEST()
651 EXPECT_EQ(d_node->NumControlledFanouts(), 2); in TYPED_TEST()
653 EXPECT_EQ(d_node->NumControlledFanouts(), 1); in TYPED_TEST()
921 expected_node_view.NumControlledFanouts(); in CompareGraphViewWithGraph()
922 EXPECT_EQ(node_view->NumControlledFanouts(), num_controlled_fanouts); in CompareGraphViewWithGraph()
Dgraph_view_internal.h217 int NumControlledFanouts() const { return controlled_fanouts_.size(); } in NumControlledFanouts() function
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer.cc285 !(node.NumRegularFanouts() == 0 && node.NumControlledFanouts() == 0); in ShouldProcess()
648 !(node.NumRegularFanouts() == 0 && node.NumControlledFanouts() == 0); in CanProcessNode()
Dremapper.cc382 node_view.NumControlledFanouts() > 0; in IsCpuCompatibleDataType()