Searched refs:step_function (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/ |
| H A D | training.py | 836 def step_function(model, iterator): function
857 return step_function(self, iterator)
864 outputs = step_function(self, iterator)
1307 def step_function(model, iterator): function
1327 return step_function(self, iterator)
1334 outputs = step_function(self, iterator)
1556 def step_function(model, iterator): function
1577 return step_function(self, iterator)
1583 outputs = step_function(self, iterator)
1589 step_outputs = step_function(self, iterator)
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/ |
| H A D | backend.py | 4172 def rnn(step_function, argument
4322 output, new_states = step_function(inp,
4358 output, states = step_function(inp, tuple(states) + tuple(constants))
4389 output_time_zero, _ = step_function(
4475 output, new_states = step_function(current_input,
4520 output, new_states = step_function(current_input,
|