Home
last modified time | relevance | path

Searched refs:new_control_inputs (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
H A Dbreakup-islands.cc65 new_control_inputs);
117 llvm::DenseMap<Operation*, llvm::SmallVector<Value, 4>> new_control_inputs; in runOnFunction() local
123 BreakUpIsland(island, side_effect_analysis, &new_control_inputs); in runOnFunction()
135 auto it = new_control_inputs.find(&item); in runOnFunction()
136 if (it == new_control_inputs.end()) continue; in runOnFunction()
272 new_control_inputs) { in BreakUpIsland() argument
366 (*new_control_inputs)[other_island_op].push_back(sink_island_control); in BreakUpIsland()
370 (*new_control_inputs)[owner].push_back(sink_island_control); in BreakUpIsland()
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dsubscribe.py141 new_control_inputs = consumer_op.control_inputs
142 if tensor.op in new_control_inputs:
143 new_control_inputs.remove(tensor.op)
144 new_control_inputs.append(out.op)
147 consumer_op._add_control_inputs(new_control_inputs)
H A Dops.py3955 new_control_inputs = self._control_dependencies_for_inputs(op.inputs)
3956 op._add_control_inputs(new_control_inputs)