Searched refs:pred_in (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/graph/ |
| D | mkl_layout_pass.cc | 2713 gtl::InlinedVector<std::pair<Node*, int>, 4> pred_in(pred_num); in MergeConv2DWithBiasAdd() local
2714 FillInputs(pred, &pred_control_edges, &pred_in); in MergeConv2DWithBiasAdd()
2741 nb.Input(pred_in[0].first, pred_in[0].second); // In1 of Conv2D in MergeConv2DWithBiasAdd()
2743 nb.Input(pred_in[1].first, pred_in[1].second); // In2 of Conv2D in MergeConv2DWithBiasAdd()
2860 gtl::InlinedVector<std::pair<Node*, int>, 4> pred_in(pred_num); in MergePadWithConv2D() local
2861 FillInputs(pred, &pred_control_edges, &pred_in); in MergePadWithConv2D()
2904 nb.Input(pred_in[0].first, pred_in[0].second); // In1 (input data) of Pad in MergePadWithConv2D()
2916 nb.Input(pred_in[1].first, pred_in[1].second); // In2 (paddings) of Pad in MergePadWithConv2D()
2921 nb.Input(pred_in[1].first, pred_in[1].second); // In2 (paddings) of Pad in MergePadWithConv2D()
|