Home
last modified time | relevance | path

Searched refs:ctrl_inputs (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/debug/lib/
Ddebug_graphs.py359 ctrl_inputs = self._node_ctrl_inputs[node]
361 for ctrl_inp in ctrl_inputs:
365 ctrl_inputs.remove(debug_op_inp)
386 ctrl_inputs = self._node_ctrl_inputs[node]
387 for ctrl_inp in ctrl_inputs:
/external/tensorflow/tensorflow/core/grappler/optimizers/
Ddependency_optimizer.cc295 std::unordered_set<string> ctrl_inputs; in OptimizeNode() local
300 if (!ctrl_inputs.insert(old_input).second) { in OptimizeNode()
312 ctrl_inputs.insert(ctrl_input); in OptimizeNode()
Darithmetic_optimizer.cc1497 std::set<string> ctrl_inputs; in TrySimplify() local
1500 FindCommonUnaryOpChain(*node, &prefix_length, &tails, &ctrl_inputs)); in TrySimplify()
1503 HoistUnaryOpChain(prefix_length, tails, &ctrl_inputs, node)); in TrySimplify()
1525 std::set<string>* ctrl_inputs) const { in FindCommonUnaryOpChain()
1537 ctrl_inputs->clear(); in FindCommonUnaryOpChain()
1544 GatherControlInputs(ctrl_inputs, *tails); in FindCommonUnaryOpChain()
1555 std::set<string>* ctrl_inputs, NodeDef* root_node) { in HoistUnaryOpChain() argument
1559 << absl::StrJoin(*ctrl_inputs, ", ") << "]"; in HoistUnaryOpChain()
1567 AddControlInputs(ctrl_inputs, root_node); in HoistUnaryOpChain()
1570 return HoistChainForSplit(prefix_length, tails, ctrl_inputs, root_node); in HoistUnaryOpChain()
[all …]
/external/tensorflow/tensorflow/python/debug/cli/
Danalyzer_cli.py800 ctrl_inputs = self._exclude_denylisted_ops(
802 output.extend(self._format_neighbors("input", inputs, ctrl_inputs))
1411 ctrl_inputs = self._exclude_denylisted_ops(
1413 all_inputs.extend(ctrl_inputs)
1414 is_ctrl.extend([True] * len(ctrl_inputs))