Searched refs:apply_fn (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/framework/ |
| D | subscribe.py | 30 def _recursive_apply(tensors, apply_fn): argument
50 return apply_fn(tensors)
52 return apply_fn(tensors.value())
54 tensors = [_recursive_apply(t, apply_fn) for t in tensors]
61 return dict((k, _recursive_apply(v, apply_fn)) for k, v in tensors.items())
|
| /external/tensorflow/tensorflow/python/autograph/pyct/ |
| D | transformer.py | 374 def apply_to_single_assignments(self, targets, values, apply_fn): argument
416 self.apply_to_single_assignments(target_el, value_el, apply_fn)
419 apply_fn(target, values)
|
| D | ast_util.py | 236 def apply_to_single_assignments(targets, values, apply_fn): argument
278 apply_to_single_assignments(target_el, value_el, apply_fn)
280 apply_fn(target, values)
|
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scale_optimizer.py | 219 def apply_fn(): function
223 maybe_apply_op = smart_cond.smart_cond(should_apply_grads, apply_fn,
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | readers.py | 580 def apply_fn(dataset): function
590 dataset = dataset.apply(apply_fn)
1054 def apply_fn(dataset): function
1064 dataset = dataset.apply(apply_fn)
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| D | map_vectorization_test.py | 526 def apply_fn(dataset): function
530 return apply_fn
533 def apply_fn(dataset): function
536 return apply_fn
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | loss_scale_optimizer.py | 723 def apply_fn(): function
744 maybe_apply_op = smart_cond.smart_cond(should_apply_grads, apply_fn,
|