Home
last modified time | relevance | path

Searched refs:GetRegularFanout (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer.cc251 for (const auto& regular_fanout : node->GetRegularFanout(0)) { in EraseCancellableNodes()
328 for (auto& fanout : pad->GetRegularFanout(0)) { in EraseCancellableNodesAroundPad()
Dremapper.cc387 return node_view.GetRegularFanout(0).size() <= 1; in IsCpuCompatibleDataType()
398 return absl::c_count_if(node_view.GetRegularFanout(0), predicate) <= 1; in IsCpuCompatibleDataType()
564 !node_view->GetRegularFanout(1).empty() || // batch_mean in IsCpuCompatibleDataType()
565 !node_view->GetRegularFanout(2).empty() || // batch_variance in IsCpuCompatibleDataType()
566 !node_view->GetRegularFanout(3).empty() || // reserve_space_1 in IsCpuCompatibleDataType()
567 !node_view->GetRegularFanout(4).empty()) // reserve_space_2 in IsCpuCompatibleDataType()
Dgeneric_layout_optimizer_test.cc212 for (const auto& regular_fanout : node->GetRegularFanout(port)) { in VerifyRegularFanoutMatch()
Dgeneric_layout_optimizer_transposer.cc421 const auto& fanouts_src_port = src_node->GetRegularFanout(src_port); in UpdateFanoutEdgesWithOp()
/external/tensorflow/tensorflow/core/grappler/utils/
Dgraph_view_test.cc476 TYPED_TEST(TypedFanoutTest, GetRegularFanout) { in TYPED_TEST() argument
493 const auto& d_fanouts_2 = d_node->GetRegularFanout(2); in TYPED_TEST()
498 const auto& d_fanouts_3 = d_node->GetRegularFanout(3); in TYPED_TEST()
509 EXPECT_EQ(d_node->GetRegularFanout(-2), no_fanouts); in TYPED_TEST()
510 EXPECT_EQ(d_node->GetRegularFanout(Graph::kControlSlot), no_fanouts); in TYPED_TEST()
511 EXPECT_EQ(d_node->GetRegularFanout(0), no_fanouts); in TYPED_TEST()
512 EXPECT_EQ(d_node->GetRegularFanout(1), no_fanouts); in TYPED_TEST()
513 EXPECT_EQ(d_node->GetRegularFanout(4), no_fanouts); in TYPED_TEST()
514 EXPECT_EQ(d_node->GetRegularFanout(5), no_fanouts); in TYPED_TEST()
2569 node->GetRegularFanout(0); in BM_GraphViewTGetRegularFanout()
Dgraph_view_internal.h194 const std::vector<FaninViewT>& GetRegularFanout(int i) const { in GetRegularFanout() function