Searched defs:apply_fn (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/python/framework/ |
| D | subscribe.py | 26 def _recursive_apply(tensors, apply_fn): argument
|
| /external/tensorflow/tensorflow/python/autograph/pyct/ |
| D | ast_util.py | 232 def apply_to_single_assignments(targets, values, apply_fn): argument
|
| D | transformer.py | 370 def apply_to_single_assignments(self, targets, values, apply_fn): argument
|
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scale_optimizer.py | 216 def apply_fn(): function
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | readers.py | 599 def apply_fn(dataset): function
1054 def apply_fn(dataset): function
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | loss_scale_optimizer.py | 636 def apply_fn(): function
649 def apply_fn(): function
|
| /external/pytorch/torch/testing/_internal/ |
| D | common_nn.py | 3858 def apply_fn(input, target, *params): function
3863 def apply_fn(input1, input2, target, *params): # type: ignore[misc] function
|
| D | common_utils.py | 4774 def _assertGradAndGradgradChecks(test_case, apply_fn, inputs, **kwargs): argument
|
| /external/pytorch/torch/csrc/autograd/ |
| D | python_function.cpp | 147 THPObjectPtr apply_fn(PyObject_GetAttrString(obj, "apply")); in apply() local
|
| /external/pytorch/aten/src/ATen/native/cpu/ |
| D | GridSamplerKernel.cpp | 1024 const TensorAccessor<const scalar_t, 3>& grid_slice, const ApplyFn &apply_fn) { in grid_sample_2d_grid_slice_iterator()
|
| /external/cronet/tot/third_party/rust/chromium_crates_io/vendor/fend-core-1.5.5/src/num/ |
| D | unit.rs | 559 fn apply_fn<I: Interrupt>( in apply_fn() method
|
| /external/cronet/stable/third_party/rust/chromium_crates_io/vendor/fend-core-1.5.5/src/num/ |
| D | unit.rs | 559 fn apply_fn<I: Interrupt>( in apply_fn() method
|
| /external/pytorch/test/distributions/ |
| D | test_distributions.py | 1227 def apply_fn(s, *params): function
|