Searched refs:_custom_step (Results 1 – 1 of 1) sorted by relevance
258 def _custom_step(features, labels): function270 strategy.run(_custom_step, args=(next(iterator))))