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