Home
last modified time | relevance | path

Searched refs:test_per_sample_weights (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/test/nn/
H A Dtest_embedding.py1005 def test_per_sample_weights(mode, trainable_scale): function
1051 test_per_sample_weights(mode, trainable)
1069 def test_per_sample_weights(mode, trainable_scale): function
1112 test_per_sample_weights(mode, trainable)
1215 test_per_sample_weights=False, argument
1230 if test_per_sample_weights:
1246 if test_per_sample_weights:
1253 assert not test_per_sample_weights
1256 assert not test_per_sample_weights
1280 if test_per_sample_weights and trainable_per_sample_weights:
[all …]