Matt 96d833b211 Return scalar losses instead of per-sample means (#18013) ... * Return scalar losses instead of per-sample means * Make loss shape (1,) instead of scalar * Allow scalar losses in test_loss_computation * Allow scalar losses in test_loss_computation * Allow scalar losses in test_loss_computation * Remove XLA loss function for RAG		2022-07-04 17:26:19 +01:00
..
__init__.py	Move test model folders (#17034)	2022-05-03 14:42:02 +02:00
test_modeling_tf_xlnet.py	Return scalar losses instead of per-sample means (#18013)	2022-07-04 17:26:19 +01:00
test_modeling_xlnet.py	Fx support for multiple model architectures (#17393)	2022-05-31 10:02:55 +02:00
test_tokenization_xlnet.py	Move test model folders (#17034)	2022-05-03 14:42:02 +02:00