Searched refs:check_batch_axis (Results 1 – 2 of 2) sorted by relevance
553 check_batch_axis=True, argument664 if not check_batch_axis:
2367 check_batch_axis=False, # Don't enforce the batch size.2426 check_batch_axis=False, # Don't enforce the batch size.