Searched refs:check_batch_axis (Results 1 – 2 of 2) sorted by relevance
239 check_batch_axis=True, argument339 if not check_batch_axis:
2510 check_batch_axis=False, # Don't enforce the batch size.2552 check_batch_axis=False, # Don't enforce the batch size.