Home
last modified time | relevance | path

Searched refs:step_function (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/contrib/distribute/python/
Dmonitor_test.py66 step_function, _ = single_loss_example(
71 _ = monitor_lib.Monitor(step_function, sess)
76 step_function, _ = single_loss_example(
81 _ = monitor_lib.Monitor(step_function, session=None)
/external/tensorflow/tensorflow/python/keras/
Dbackend_test.py1021 def step_function(x, states): function
1027 return step_function
1111 def step_function(x, states): function
1118 return step_function
1209 def step_function(inputs, states): function
1238 step_function,
1254 def step_function(inputs, states): function
1276 step_function,
1289 def step_function(inputs, states): function
1306 step_function,
Dbackend.py3297 def rnn(step_function, argument
3442 output, new_states = step_function(inp,
3478 output, states = step_function(inp, tuple(states) + tuple(constants))
3509 output_time_zero, _ = step_function(input_time_zero,
3564 output, new_states = step_function(current_input,
3612 output, new_states = step_function(current_input,
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.keras.backend.pbtxt417 …argspec: "args=[\'step_function\', \'inputs\', \'initial_states\', \'go_backwards\', \'mask\', \'c…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.keras.backend.pbtxt421 …argspec: "args=[\'step_function\', \'inputs\', \'initial_states\', \'go_backwards\', \'mask\', \'c…