| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer.cc | 238 const auto& regular_fanin_0 = node->GetRegularFanin(0); in EraseCancellableNodes()
248 const auto& fanin_to_forward = fanin_node->GetRegularFanin(0); in EraseCancellableNodes()
258 mutation->RemoveNode(node->GetRegularFanin(1).node_view()); in EraseCancellableNodes()
262 mutation->RemoveNode(fanin_node->GetRegularFanin(1).node_view()); in EraseCancellableNodes()
290 const auto& transpose_after_fanin = transpose_after->GetRegularFanin(0); in EraseCancellableNodesAroundPad()
295 const auto& pad_fanin_0 = pad->GetRegularFanin(0); in EraseCancellableNodesAroundPad()
312 const auto& pad_fanin_1 = pad->GetRegularFanin(1); in EraseCancellableNodesAroundPad()
|
| D | remapper.cc | 414 const auto& regular_fanin_0 = node_view->GetRegularFanin(0); in IsCpuCompatibleDataType()
454 const auto& regular_fanin_0 = node_view->GetRegularFanin(0); in IsCpuCompatibleDataType()
468 bias_add_node_view->GetRegularFanin(0).node_view(); in IsCpuCompatibleDataType()
502 const auto& regular_fanin_0 = node_view->GetRegularFanin(0); in IsCpuCompatibleDataType()
522 const auto& squeeze_regular_fanin_0 = squeeze_node_view->GetRegularFanin(0); in IsCpuCompatibleDataType()
572 const auto& regular_fanin_0 = node_view->GetRegularFanin(0); in IsCpuCompatibleDataType()
606 const auto& regular_fanin_0 = node_view->GetRegularFanin(0); in IsCpuCompatibleDataType()
640 add_node_view.GetRegularFanin(port_id).node_view(); in IsCpuCompatibleDataType()
744 const auto& regular_fanin_0 = node_view->GetRegularFanin(0); in IsCpuCompatibleDataType()
755 add_node_view->GetRegularFanin(base.port_id).node_view(); in IsCpuCompatibleDataType()
[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 | 381 auto& fanin_port = dst_node->GetRegularFanin(dst_port); in UpdateFaninEdgesWithOp()
591 const auto& regular_fanin = node.GetRegularFanin(port); in GetFaninPortRank()
606 const auto& regular_fanin = node.GetRegularFanin(port); in IsFaninPortDimsNIfConst()
849 const auto& fanin = node->GetRegularFanin(0); in TransposeNode()
996 const auto& data_fanin = node->GetRegularFanin(0); in TransposeNode()
1105 const auto& fanin = node.GetRegularFanin(pos); in IsAfterDstToSrcTransform()
1123 const auto& fanin = current_node->GetRegularFanin(pos); in IsAfterDstToSrcTransform()
1141 const auto& regular_fanin = node.GetRegularFanin(i); in GetVariadicNDFaninPorts()
1294 const auto& fanin = node->GetRegularFanin(vector_index); in MaybeReshapeVectorFanin()
1509 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 | 1834 graph_view_.GetRegularFanin(node_input); in NodeImplicitlyReadsNonResourceVariable()
|
| 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 | 428 const MutableGraphView::OutputPort MutableGraphView::GetRegularFanin( in GetRegularFanin() function in tensorflow::grappler::MutableGraphView
430 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 | 79 return graph.GetRegularFanin(input_port).node; in GetInputNode()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | graph_utils.cc | 274 return graph.GetRegularFanin(input_port).node; in GetInputNode()
281 return graph.GetRegularFanin(input_port).node; in GetInputNode()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 976 const GraphView::OutputPort fanin = graph_.GetRegularFanin(port); in UpdateNode()
2278 GraphView::OutputPort fanin = shape_refiner->graph().GetRegularFanin(port); in UpdateEnter()
2450 GraphView::OutputPort fanin = shape_refiner->graph().GetRegularFanin(inp); in UpdateEnqueue()
2637 GraphView::OutputPort fanin = graph_view.GetRegularFanin(input); in InferStatically()
|