Home
last modified time | relevance | path

Searched defs:step_fn (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dzero_batch_test.py58 def step_fn(is_training, inputs, targets=None): function
129 def step_fn(inputs, targets): function
152 def step_fn(inputs): function
Dminimize_loss_test.py95 def step_fn(ctx, inputs): function
190 def step_fn(ctx, inputs): function
255 def step_fn(ctx, inputs): function
365 def step_fn(ctx, inputs): function
453 def step_fn(output_context, inputs): function
Dcustom_training_loop_test.py282 def step_fn(data): function
551 def step_fn(data): function
701 def step_fn(inputs): function
730 def step_fn(inputs): function
760 def step_fn(inputs): function
803 def step_fn(inputs): function
834 def step_fn(inputs): function
884 def step_fn(inputs): function
941 def step_fn(inputs): function
958 def step_fn(inputs): function
Dctl_correctness_test.py113 def step_fn(inputs): function
176 def step_fn(inputs): function
Dstep_fn.py97 def step_fn(ctx, inputs): function
Dmetrics_v1_test.py104 def step_fn(ctx, inputs): function
Dmirrored_strategy_test.py1346 def step_fn(): function
/external/tensorflow/tensorflow/python/training/
Dmonitored_session_test.py1052 def step_fn(step_context): function
1085 def step_fn(step_context): function
1118 def step_fn(step_context): function
1164 def step_fn(step_context): function
1200 def step_fn(step_context): function
1235 def step_fn(step_context): function
1916 def step_fn(step_context): function
1926 def step_fn(step_context): function
1937 def step_fn(step_context): function
1951 def step_fn(step_context): function
[all …]
Dmonitored_session.py780 def run_step_fn(self, step_fn): argument
1202 def run_step_fn(self, step_fn, raw_session, run_with_hooks): argument
1297 def run_step_fn(self, step_fn, raw_session, run_with_hooks): argument
/external/tensorflow/tensorflow/python/debug/wrappers/
Dframework.py644 def run_step_fn(self, step_fn): argument
/external/tensorflow/tensorflow/python/keras/distribute/
Ddistribute_strategy_test.py1971 def step_fn(inputs): function