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.py360 ctrl_inputs = self._node_ctrl_inputs[node]
362 for ctrl_inp in ctrl_inputs:
366 ctrl_inputs.remove(debug_op_inp)
387 ctrl_inputs = self._node_ctrl_inputs[node]
388 for ctrl_inp in ctrl_inputs:
/external/tensorflow/tensorflow/core/grappler/optimizers/
Ddependency_optimizer.cc273 std::unordered_set<string> ctrl_inputs; in OptimizeNode() local
278 if (!ctrl_inputs.insert(old_input).second) { in OptimizeNode()
290 ctrl_inputs.insert(ctrl_input); in OptimizeNode()
Darithmetic_optimizer.cc1450 std::set<string> ctrl_inputs; in TrySimplify() local
1453 FindCommonUnaryOpChain(*node, &prefix_length, &tails, &ctrl_inputs)); in TrySimplify()
1456 HoistUnaryOpChain(prefix_length, tails, &ctrl_inputs, node)); in TrySimplify()
1466 std::set<string>* ctrl_inputs) const { in FindCommonUnaryOpChain()
1478 ctrl_inputs->clear(); in FindCommonUnaryOpChain()
1485 GatherControlInputs(ctrl_inputs, *tails); in FindCommonUnaryOpChain()
1496 std::set<string>* ctrl_inputs, NodeDef* root_node) { in HoistUnaryOpChain() argument
1503 AddControlInputs(ctrl_inputs, root_node); in HoistUnaryOpChain()
1506 return HoistChainForSplit(prefix_length, tails, ctrl_inputs, root_node); in HoistUnaryOpChain()
1510 void GatherControlInputs(std::set<string>* ctrl_inputs, in GatherControlInputs() argument
[all …]
/external/tensorflow/tensorflow/python/debug/cli/
Danalyzer_cli.py802 ctrl_inputs = self._exclude_blacklisted_ops(
804 output.extend(self._format_neighbors("input", inputs, ctrl_inputs))
1412 ctrl_inputs = self._exclude_blacklisted_ops(
1414 all_inputs.extend(ctrl_inputs)
1415 is_ctrl.extend([True] * len(ctrl_inputs))