Home
last modified time | relevance | path

Searched defs:fanin (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dstatic_schedule.cc90 const NodeDef* fanin = it->second; in EstimateEarliestExecutionTimes() local
153 const NodeDef* fanin = it->second; in EstimateRequiredTimes() local
184 const NodeDef* fanin = name_map[NodeName(fanin_name)]; in EstimateRequiredTimes() local
Dpin_to_host_optimizer.cc125 for (const auto& fanin : graph.GetFanins(node, false)) { in IsNodeOutputPortHostFriendly() local
244 for (const GraphView::OutputPort& fanin : in IsNodeHostCandidate() local
Dgeneric_layout_optimizer_transposer.cc849 const auto& fanin = node->GetRegularFanin(0); in TransposeNode() local
1105 const auto& fanin = node.GetRegularFanin(pos); in IsAfterDstToSrcTransform() local
1123 const auto& fanin = current_node->GetRegularFanin(pos); in IsAfterDstToSrcTransform() local
1294 const auto& fanin = node->GetRegularFanin(vector_index); in MaybeReshapeVectorFanin() local
1582 const utils::MutableNodeView& fanin, int port) { in IsFaninScalarVector4D()
1588 const utils::MutableNodeView& fanin, int port) { in GetFaninPorts()
Dmemory_optimizer.cc867 for (const string& fanin : input_node->input()) { in FindSwapInTrigger() local
915 MutableGraphView::OutputPort fanin = graph.GetRegularFanin(input); in IsSwappable() local
Dgeneric_layout_optimizer_test.cc204 const auto& fanin = node->GetRegularFanin(port); in VerifyRegularFaninMatch() local
Dauto_mixed_precision.cc811 for (const int fanin : graph_type_view.GetFanin(w.node)) { in DfsTypeTraversal() local
Dgeneric_layout_optimizer_transposer_test.cc77 const auto& fanin = node->GetRegularFanin(port); in VerifyRegularFaninMatch() local
/external/tensorflow/tensorflow/core/grappler/
Dmutable_graph_view.cc267 Status CheckFaninIsRegular(const TensorId& fanin, ErrorHandler handler) { in CheckFaninIsRegular()
275 Status CheckFaninIsValid(const TensorId& fanin, ErrorHandler handler) { in CheckFaninIsValid()
284 const TensorId& fanin, ErrorHandler handler) { in CheckAddingFaninToSelf()
293 const TensorId& fanin, ErrorHandler handler) { in CheckRemovingFaninFromSelf()
331 const MutableGraphView::OutputPort& fanin) { in GeneratedNameForIdentityConsumingSwitch()
389 const OutputPort& fanin, in UpdateMaxRegularOutputPortForRemovedFanin()
410 const OutputPort& fanin) { in UpdateMaxRegularOutputPortForAddedFanin()
846 const OutputPort& fanin) { in AddFaninInternal()
894 const TensorId& fanin) { in AddRegularFanin()
914 const TensorId& fanin) { in AddRegularFaninByPort()
[all …]
Dgrappler_item.cc95 std::vector<const NodeDef*> fanin; in MainVariables() local
Dgraph_view_test.cc137 GraphView::OutputPort fanin = graph.GetRegularFanin(input); in TEST_F() local
211 auto fanin = graph.GetFanin(input); in TEST_F() local
Dmutable_graph_view_test.cc115 for (const auto& fanin : mutable_fanins) { in CheckGraph() local
1184 void CheckFanoutRemoved(const MutableGraphView& graph, const TensorId& fanin, in CheckFanoutRemoved()
1470 for (const string& fanin : fanin_strings) { in TestRemoveAllFanins() local
1681 void TestUpdateFaninFromFaninToNodeAsSwitchControl(const TensorId& fanin) { in TestUpdateFaninFromFaninToNodeAsSwitchControl()
1716 int port, const TensorId& fanin, bool success, in TestUpdateRegularFaninByPort()
2550 const TensorId& fanin, in TestAddControllingFaninSelfLoops()
Dgraph_view.h182 bool HasFanin(const NodeDefT& node, const TensorId& fanin) const { in HasFanin()
/external/tensorflow/tensorflow/core/grappler/utils/
Dgraph_view.cc332 const TensorId& fanin) { in AddOrUpdateRegularFanin()
339 const TensorId& fanin) { in AddOrUpdateRegularFanin()
446 const internal::NodeDefAndPortIndex& fanin) { in IncrementFaninCount()
452 const internal::NodeDefAndPortIndex& fanin) { in DecrementFaninCount()
997 auto& fanin = fanout_node_view->regular_fanins_[fanout.index()]; in FixRenamedFanouts() local
1005 auto& fanin = fanout_node_view->controlling_fanins_[fanout.fanin_index_]; in FixRenamedFanouts() local
1016 MutableFanoutView& fanin = node_view->regular_fanins_[i]; in RemoveRegularFaninFanoutInternal() local
1233 for (const SafeTensorId& fanin : diff.regular_inputs_to_add) { in ApplyNodeUpdates() local
1282 for (const SafeTensorId& fanin : new_node->regular_fanins) { in SetNewNodesFanins() local
Dtraversal.cc108 for (const int fanin : graph_view.GetFanin(w.node)) { in DfsTraversal() local
Dgraph_view_internal.h490 const TensorId& fanin) { in AddOrUpdateRegularFanin()
828 const TensorId& fanin) { in AddOrUpdateRegularFanin()
Dgraph_view_test.cc841 const auto& fanin = node_view->GetRegularFanin(i); in CompareGraphViewWithGraph() local
877 const auto& fanin = node_view->GetControllingFanins()[i]; in CompareGraphViewWithGraph() local
2710 auto* fanin = graph_view.GetNode("node"); in BM_GraphViewTHasRegularFanin() local
2753 auto* fanin = graph_view.GetNode("node"); in BM_GraphViewTHasControllingFanin() local
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dauto_shard.cc321 for (const auto& fanin : graph->GetFanins(node, true)) { in RemoveShuffleDataset() local
343 for (const auto& fanin : graph->GetFanins(node, true)) { in RemoveShuffleDatasetV2() local
370 for (const auto& fanin : graph->GetFanins(node, true)) { in RemoveShuffleDatasetV3() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc976 const GraphView::OutputPort fanin = graph_.GetRegularFanin(port); in UpdateNode() local
2235 for (const GraphView::Edge fanin : shape_refiner->graph().GetFaninEdges( in UpdateMerge() local
2278 GraphView::OutputPort fanin = shape_refiner->graph().GetRegularFanin(port); in UpdateEnter() local
2450 GraphView::OutputPort fanin = shape_refiner->graph().GetRegularFanin(inp); in UpdateEnqueue() local
2637 GraphView::OutputPort fanin = graph_view.GetRegularFanin(input); in InferStatically() local