| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer.cc | 245 const auto& regular_fanin_0 = node->GetRegularFanin(0); in EraseCancellableNodes()
255 const auto& fanin_to_forward = fanin_node->GetRegularFanin(0); in EraseCancellableNodes()
265 mutation->RemoveNode(node->GetRegularFanin(1).node_view()); in EraseCancellableNodes()
269 mutation->RemoveNode(fanin_node->GetRegularFanin(1).node_view()); in EraseCancellableNodes()
297 const auto& transpose_after_fanin = transpose_after->GetRegularFanin(0); in EraseCancellableNodesAroundPad()
302 const auto& pad_fanin_0 = pad->GetRegularFanin(0); in EraseCancellableNodesAroundPad()
319 const auto& pad_fanin_1 = pad->GetRegularFanin(1); in EraseCancellableNodesAroundPad()
|
| D | remapper.cc | 589 const auto& regular_fanin_0 = node_view.GetRegularFanin(0); in IsBiasSemanticAdd()
592 const auto& regular_fanin_1 = node_view.GetRegularFanin(1); in IsBiasSemanticAdd()
673 const auto& regular_fanin_0 = node_view->GetRegularFanin(1 - bias_port); in FindContractionWithBias()
714 const auto& regular_fanin_0 = node_view->GetRegularFanin(0); in FindContractionWithBiasAndActivation()
728 bias_add_node_view->GetRegularFanin(1 - base.bias_port).node_view(); in FindContractionWithBiasAndActivation()
765 const auto& regular_fanin_0 = node_view->GetRegularFanin(0); in FindConvWithSqueezeAndBias()
778 const auto& squeeze_regular_fanin_0 = squeeze_node_view->GetRegularFanin(0); in FindConvWithSqueezeAndBias()
843 const auto& regular_fanin_0 = node_view->GetRegularFanin(0); in FindConv2DWithBatchNorm()
881 const auto& regular_fanin_0 = node_view->GetRegularFanin(0); in FindConv2DWithBatchNormAndActivation()
914 add_node_view.GetRegularFanin(port_id).node_view(); in FindContractionWithBiasInPort()
[all …]
|
| D | shape_optimizer.cc | 91 graph.GetRegularFanin(MutableGraphView::InputPort(fanout.node, 1)); in Optimize()
149 graph.GetRegularFanin(MutableGraphView::InputPort(&node, 0)); in Optimize()
151 graph.GetRegularFanin(MutableGraphView::InputPort(&node, 1)); in Optimize()
|
| D | generic_layout_optimizer_transposer.cc | 378 auto& fanin_port = dst_node->GetRegularFanin(dst_port); in UpdateFaninEdgesWithOp()
586 const auto& regular_fanin = node.GetRegularFanin(port); in GetFaninPortRank()
601 const auto& regular_fanin = node.GetRegularFanin(port); in IsFaninPortDimsNIfConst()
844 const auto& fanin = node->GetRegularFanin(0); in TransposeNode()
991 const auto& data_fanin = node->GetRegularFanin(0); in TransposeNode()
1100 const auto& fanin = node.GetRegularFanin(pos); in IsAfterDstToSrcTransform()
1118 const auto& fanin = current_node->GetRegularFanin(pos); in IsAfterDstToSrcTransform()
1136 const auto& regular_fanin = node.GetRegularFanin(i); in GetVariadicNDFaninPorts()
1289 const auto& fanin = node->GetRegularFanin(vector_index); in MaybeReshapeVectorFanin()
1504 const auto& regular_fanin_1 = node.GetRegularFanin(1); in IsReduceAxisSupported()
[all …]
|
| D | loop_optimizer.cc | 605 view.GetRegularFanin(switch_loopcond_port).node; in CheckForDeadFanout()
620 const NodeDef* switch_input = view.GetRegularFanin(switch_input_port).node; in CheckForDeadFanout()
634 NodeDef* switch_ctrl_node = view.GetRegularFanin({switch_predicate, 0}).node; in CheckForDeadFanout()
|
| D | implementation_selector.cc | 114 backward_node->GetRegularFanin(last_input_index); in FindForwardNode()
|
| D | memory_optimizer.cc | 915 MutableGraphView::OutputPort fanin = graph.GetRegularFanin(input); in IsSwappable()
931 MutableGraphView::OutputPort generator = view.GetRegularFanin(swap); in FindSwapOutTrigger()
|
| D | generic_layout_optimizer_test.cc | 204 const auto& fanin = node->GetRegularFanin(port); in VerifyRegularFaninMatch()
|
| D | auto_mixed_precision.cc | 2014 graph_view_.GetRegularFanin(node_input); in NodeImplicitlyReadsNonResourceVariable()
2197 MutableGraphView::OutputPort src = graph_view_.GetRegularFanin(dst); in ChangeTypeAttrsAndAddCasts()
|
| D | generic_layout_optimizer_transposer_test.cc | 77 const auto& fanin = node->GetRegularFanin(port); in VerifyRegularFaninMatch()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | graph_view_test.cc | 137 GraphView::OutputPort fanin = graph.GetRegularFanin(input); in TEST_F()
144 fanin = graph.GetRegularFanin(input); in TEST_F()
285 auto d_output_0 = graph.GetRegularFanin({d_node, 0}); in TEST_F()
287 auto d_output_1 = graph.GetRegularFanin({d_node, 1}); in TEST_F()
289 auto d_output_2 = graph.GetRegularFanin({d_node, 2}); in TEST_F()
291 auto d_output_control = graph.GetRegularFanin({d_node, Graph::kControlSlot}); in TEST_F()
431 graph_view.GetRegularFanin({node, 0}); in BM_GraphViewGetRegularFanin()
|
| D | mutable_graph_view.h | 59 using GraphViewInternal::GetRegularFanin;
60 const OutputPort GetRegularFanin(const GraphView::InputPort& port) const;
|
| D | graph_view.h | 147 OutputPort regular_fanin = GetRegularFanin(port); in GetFanin()
170 const OutputPort GetRegularFanin(const InputPort& port) const { in GetRegularFanin() function
|
| D | mutable_graph_view.cc | 450 const MutableGraphView::OutputPort MutableGraphView::GetRegularFanin( in GetRegularFanin() function in tensorflow::grappler::MutableGraphView
452 return GetRegularFanin(MutableGraphView::InputPort( in GetRegularFanin()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | graph_view_test.cc | 350 TYPED_TEST(TypedFaninTest, GetRegularFanin) { in TYPED_TEST() argument
367 const auto& a_fanin_0 = a_node->GetRegularFanin(0); in TYPED_TEST()
369 const auto& a_fanin_1 = a_node->GetRegularFanin(1); in TYPED_TEST()
371 const auto& a_fanin_2 = a_node->GetRegularFanin(2); in TYPED_TEST()
373 const auto& a_fanin_3 = a_node->GetRegularFanin(3); in TYPED_TEST()
380 const auto& a_fanin_4 = a_node->GetRegularFanin(4); in TYPED_TEST()
382 const auto& a_fanin_5 = a_node->GetRegularFanin(5); in TYPED_TEST()
384 const auto& a_fanin_control = a_node->GetRegularFanin(Graph::kControlSlot); in TYPED_TEST()
386 const auto& a_fanin_bad = a_node->GetRegularFanin(-2); in TYPED_TEST()
823 const auto& expected_fanin = expected_node_view.GetRegularFanin(i); in CompareGraphViewWithGraph()
[all …]
|
| D | graph_view_internal.h | 174 const FanoutViewT& GetRegularFanin(int i) const { in GetRegularFanin() function
510 const auto& existing_fanin = node_view->GetRegularFanin(index); in AddOrUpdateRegularFanin()
730 node_view->GetRegularFanin(i).node_view()->GetName(); in IsWellFormed()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | compute_batch_size_op.cc | 81 return graph.GetRegularFanin(input_port).node; in GetInputNode()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | graph_utils.cc | 282 return graph.GetRegularFanin(input_port).node; in GetInputNode()
289 return graph.GetRegularFanin(input_port).node; in GetInputNode()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 1005 const GraphView::OutputPort fanin = graph_.GetRegularFanin(port); in UpdateNode()
2322 GraphView::OutputPort fanin = shape_refiner->graph().GetRegularFanin(port); in UpdateEnter()
2494 GraphView::OutputPort fanin = shape_refiner->graph().GetRegularFanin(inp); in UpdateEnqueue()
2681 GraphView::OutputPort fanin = graph_view.GetRegularFanin(input); in InferStatically()
|