Searched refs:old_node_inputs (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/graph/ |
| D | mkl_layout_pass.cc | 1415 const gtl::InlinedVector<std::pair<Node*, int>, 4>& old_node_inputs,
1705 const gtl::InlinedVector<std::pair<Node*, int>, 4>& old_node_inputs, in SetUpContiguousInputs() argument
1761 CHECK_GE(old_node_inputs.size(), old_node_input_slots); in SetUpContiguousInputs()
1769 CHECK_LT(iidx, old_node_inputs.size()); in SetUpContiguousInputs()
1774 GetNodesProducingTFTensorList(old_node_inputs, &iidx, N, in SetUpContiguousInputs()
1784 nb->Input(old_node_inputs[iidx].first, old_node_inputs[iidx].second); in SetUpContiguousInputs()
1808 CHECK_LT(iidx, old_node_inputs.size()); in SetUpContiguousInputs()
1813 GetNodesProducingMklTensorList(g, old_node, old_node_inputs, &iidx, N, in SetUpContiguousInputs()
1827 GetNodeProducingMklTensor(g, old_node, old_node_inputs[iidx].first, in SetUpContiguousInputs()
1828 old_node_inputs[iidx].second, &mkl_node, in SetUpContiguousInputs()
[all …]
|