Searched refs:num_repeat (Results 1 – 2 of 2) sorted by relevance
34 def _parse_example_dataset(self, num_repeat, batch_size): argument37 num_epochs=num_repeat,44 num_repeat = 546 num_outputs = self._num_records * self._num_files * num_repeat // batch_size50 num_repeat=num_repeat, batch_size=batch_size), num_outputs)
156 int num_repeat = 1; in ArgDefIndexes() local158 num_repeat = node.attr().at(arg_def.number_attr()).i(); in ArgDefIndexes()160 argdef_inds.insert(argdef_inds.end(), num_repeat, {arg_idx, -1}); in ArgDefIndexes()