Searched refs:can_modify (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/contrib/graph_editor/ |
| D | reroute.py | 101 def _reroute_t(t0, t1, consumers1, can_modify=None, cannot_modify=None): argument
120 if can_modify is not None:
121 consumers1 &= can_modify
135 def _reroute_ts(ts0, ts1, mode, can_modify=None, cannot_modify=None): argument
186 if can_modify is not None:
187 can_modify = frozenset(_util.make_list_of_op(can_modify))
200 nb_update_inputs += _reroute_t(t0, t1, consumers1, can_modify,
203 nb_update_inputs += _reroute_t(t1, t0, consumers0, can_modify,
208 def swap_ts(ts0, ts1, can_modify=None, cannot_modify=None): argument
230 return _reroute_ts(ts0, ts1, _RerouteMode.swap, can_modify, cannot_modify)
[all …]
|
| /external/tensorflow/tensorflow/contrib/quantize/python/ |
| D | common.py | 136 def RerouteTensor(t0, t1, can_modify=None): argument
150 if can_modify is not None:
151 consumers = [c for c in consumers if c in can_modify]
|
| D | fold_batch_norms.py | 419 can_modify=bn_decay_mean_consumers)
430 can_modify=bn_decay_var_consumers)
536 folded_op.outputs[0], original_op.outputs[0], can_modify=[activation])
561 folded_op.outputs[0], original_op.outputs[0], can_modify=[add_bypass])
|
| D | common_test.py | 87 common.RerouteTensor(b, a, can_modify=[add_ac.op])
|
| D | quantize.py | 769 quant, inputs, can_modify=consumers)
|