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.py355 ctrl_inputs = self._node_ctrl_inputs[node]
357 for ctrl_inp in ctrl_inputs:
361 ctrl_inputs.remove(debug_op_inp)
382 ctrl_inputs = self._node_ctrl_inputs[node]
383 for ctrl_inp in ctrl_inputs:
/external/tensorflow/tensorflow/python/debug/cli/
Danalyzer_cli.py795 ctrl_inputs = self._exclude_denylisted_ops(
797 output.extend(self._format_neighbors("input", inputs, ctrl_inputs))
1406 ctrl_inputs = self._exclude_denylisted_ops(
1408 all_inputs.extend(ctrl_inputs)
1409 is_ctrl.extend([True] * len(ctrl_inputs))
/external/tensorflow/tensorflow/core/grappler/optimizers/
Ddependency_optimizer.cc303 std::unordered_set<string> ctrl_inputs; in OptimizeNode() local
308 if (!ctrl_inputs.insert(old_input).second) { in OptimizeNode()
320 ctrl_inputs.insert(ctrl_input); in OptimizeNode()
Darithmetic_optimizer.cc1559 std::set<string> ctrl_inputs; in TrySimplify() local
1562 FindCommonUnaryOpChain(*node, &prefix_length, &tails, &ctrl_inputs)); in TrySimplify()
1565 HoistUnaryOpChain(prefix_length, tails, &ctrl_inputs, node)); in TrySimplify()
1587 std::set<string>* ctrl_inputs) const { in FindCommonUnaryOpChain()
1599 ctrl_inputs->clear(); in FindCommonUnaryOpChain()
1606 GatherControlInputs(ctrl_inputs, *tails); in FindCommonUnaryOpChain()
1617 std::set<string>* ctrl_inputs, NodeDef* root_node) { in HoistUnaryOpChain() argument
1621 << absl::StrJoin(*ctrl_inputs, ", ") << "]"; in HoistUnaryOpChain()
1629 AddControlInputs(ctrl_inputs, root_node); in HoistUnaryOpChain()
1632 return HoistChainForSplit(prefix_length, tails, ctrl_inputs, root_node); in HoistUnaryOpChain()
[all …]