Searched refs:pred_in (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 3017 gtl::InlinedVector<std::pair<Node*, int>, 4> pred_in(pred_num); in MergeConv2DWithBiasAdd() local
3018 FillInputs(pred, &pred_control_edges, &pred_in); in MergeConv2DWithBiasAdd()
3045 nb.Input(pred_in[0].first, pred_in[0].second); // In1 of Conv2D in MergeConv2DWithBiasAdd()
3047 nb.Input(pred_in[1].first, pred_in[1].second); // In2 of Conv2D in MergeConv2DWithBiasAdd()
3178 gtl::InlinedVector<std::pair<Node*, int>, 4> pred_in(pred_num); in MergePadWithConv2D() local
3179 FillInputs(pred, &pred_control_edges, &pred_in); in MergePadWithConv2D()
3222 nb.Input(pred_in[0].first, pred_in[0].second); // In1 (input data) of Pad in MergePadWithConv2D()
3234 nb.Input(pred_in[1].first, pred_in[1].second); // In2 (paddings) of Pad in MergePadWithConv2D()
3239 nb.Input(pred_in[1].first, pred_in[1].second); // In2 (paddings) of Pad in MergePadWithConv2D()
|