Searched refs:inputs_needed (Results 1 – 2 of 2) sorted by relevance
826 std::unordered_map<const NodeDef*, int> inputs_needed; in TopologicalSortNodesWithTimePriority() local836 inputs_needed[ndef] = ndef->input_size(); in TopologicalSortNodesWithTimePriority()853 --inputs_needed[n]; in TopologicalSortNodesWithTimePriority()873 if (--inputs_needed[n] == 0) { in TopologicalSortNodesWithTimePriority()
1956 const int inputs_needed = op_info.op() == "Slice" ? 3 : 2; in PredictGatherOrSlice() local1957 if (op_info.outputs_size() == 0 || op_info.inputs_size() < inputs_needed) { in PredictGatherOrSlice()