Searched refs:scan_func (Results 1 - 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/data/python/ops/ |
| H A D | scan_ops.py | 33 def __init__(self, input_dataset, initial_state, scan_func):
89 ret = scan_func(old_state, input_value)
173 def scan(initial_state, scan_func):
177 In addition to mapping `scan_func` across the elements of the input dataset,
184 scan_func: A function that maps `(old_state, input_element)` to
194 return _ScanDataset(dataset, initial_state, scan_func)
|
| /external/tensorflow/tensorflow/contrib/integrate/python/ops/ |
| H A D | odes.py | 536 scan_func = self._make_scan_func(evol_func)
538 y_grid = functional_ops.scan(scan_func, (time_grid[:-1], time_delta_grid),
544 def scan_func(y, t_and_dt): function in function:_FixedGridIntegrator._make_scan_func
550 return scan_func
|
Completed in 580 milliseconds