Searched refs:GetRegularFanin (Results 1 – 11 of 11) sorted by relevance
57 graph.GetRegularFanin(MutableGraphView::InputPort(fanout.node, 1)); in Optimize()94 graph.GetRegularFanin(MutableGraphView::InputPort(&node, 0)); in Optimize()96 graph.GetRegularFanin(MutableGraphView::InputPort(&node, 1)); in Optimize()
241 const auto conv2d = ctx.graph_view.GetRegularFanin(input_port); in FindConv2DWithBias()272 const auto bias_add = ctx.graph_view.GetRegularFanin(input_port); in FindConv2DWithBiasAndRelu()304 const auto squeeze = ctx.graph_view.GetRegularFanin(bias_input_port); in FindConv2DWithSqueezeAndBias()322 const auto conv2d = ctx.graph_view.GetRegularFanin(squeeze_input_port); in FindConv2DWithSqueezeAndBias()369 const auto conv2d = ctx.graph_view.GetRegularFanin(input_port); in FindConv2DWithBatchNorm()401 const auto batch_norm = ctx.graph_view.GetRegularFanin(input_port); in FindConv2DWithBatchNormAndRelu()
603 view.GetRegularFanin(switch_loopcond_port).node; in CheckForDeadFanout()614 const NodeDef* switch_input = view.GetRegularFanin(switch_input_port).node; in CheckForDeadFanout()
884 MutableGraphView::OutputPort fanin = graph.GetRegularFanin(input); in IsSwappable()900 MutableGraphView::OutputPort generator = view.GetRegularFanin(swap); in FindSwapOutTrigger()
1511 ctx->graph_view().GetRegularFanin({&func_node, input_idx}); in PlaceInlinedFunctionBody()
136 GraphView::OutputPort fanin = graph.GetRegularFanin(input); in TEST_F()143 fanin = graph.GetRegularFanin(input); in TEST_F()284 auto d_output_0 = graph.GetRegularFanin({d_node, 0}); in TEST_F()286 auto d_output_1 = graph.GetRegularFanin({d_node, 1}); in TEST_F()288 auto d_output_2 = graph.GetRegularFanin({d_node, 2}); in TEST_F()290 auto d_output_control = graph.GetRegularFanin({d_node, Graph::kControlSlot}); in TEST_F()429 graph_view.GetRegularFanin({node, 0}); in BM_GraphViewGetRegularFanin()
59 using GraphViewInternal::GetRegularFanin;60 const OutputPort GetRegularFanin(const GraphView::InputPort& port) const;
147 OutputPort regular_fanin = GetRegularFanin(port); in GetFanin()170 const OutputPort GetRegularFanin(const InputPort& port) const { in GetRegularFanin() function
428 const MutableGraphView::OutputPort MutableGraphView::GetRegularFanin( in GetRegularFanin() function in tensorflow::grappler::MutableGraphView430 return GetRegularFanin(MutableGraphView::InputPort( in GetRegularFanin()
232 return graph.GetRegularFanin(input_port).node; in GetInputNode()239 return graph.GetRegularFanin(input_port).node; in GetInputNode()
763 const GraphView::OutputPort fanin = graph_.GetRegularFanin(port); in UpdateNode()1845 GraphView::OutputPort fanin = shape_refiner->graph().GetRegularFanin(port); in UpdateEnter()2017 GraphView::OutputPort fanin = shape_refiner->graph().GetRegularFanin(inp); in UpdateEnqueue()2197 GraphView::OutputPort fanin = graph_view.GetRegularFanin(input); in InferStatically()