Searched refs:seed_ops (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | op_selector.py | 237 def get_backward_walk_ops(seed_ops, argument
269 if not is_iterable(seed_ops):
270 seed_ops = [seed_ops]
271 if not seed_ops:
273 if isinstance(seed_ops[0], ops.Tensor):
274 ts = make_list_of_t(seed_ops, allow_graph=False)
275 seed_ops = get_generating_ops(ts)
277 seed_ops = make_list_of_op(seed_ops, allow_graph=False)
280 seed_ops = object_identity.ObjectIdentitySet(make_list_of_op(seed_ops))
284 seed_ops &= within_ops
[all …]
|
| D | op_selector_test.py | 124 seed_ops = [self.h.op]
136 seed_ops,
145 seed_ops,
154 seed_ops,
162 seed_ops, inclusive=True, within_ops=within_ops)
170 ops = op_selector.get_backward_walk_ops(seed_ops, inclusive=True)
|
| D | custom_gradient.py | 301 seed_ops=output_ops,
|