| /external/pytorch/test/onnx/ |
| D | test_onnxscript_runtime.py | 54 def test_layer_norm(self): member in TestONNXScriptRuntime
|
| D | test_pytorch_onnx_onnxruntime.py | 3993 def test_layer_norm(self): member in TestONNXRuntime
|
| /external/pytorch/test/functorch/ |
| D | test_memory_efficient_fusion.py | 144 def test_layer_norm(self): member in TestMemoryEfficientOpAuthoring
|
| /external/pytorch/aten/src/ATen/test/ |
| D | vulkan_api_test.cpp | 2837 void test_layer_norm( in test_layer_norm() function
2871 test_layer_norm({5, 7}, {7}, {7}, {7}, 1e-05); in TEST_F()
2872 test_layer_norm({5, 7}, {5, 7}, {5, 7}, {5, 7}, 1e-05); in TEST_F()
2876 test_layer_norm({11, 5, 7}, {7}, {7}, {7}, 1e-05); in TEST_F()
2877 test_layer_norm({11, 5, 7}, {5, 7}, {5, 7}, {5, 7}, 1e-05); in TEST_F()
2878 test_layer_norm({11, 5, 7}, {11, 5, 7}, {11, 5, 7}, {11, 5, 7}, 1e-05); in TEST_F()
2882 test_layer_norm({3, 11, 5, 7}, {7}, {7}, {7}, 1e-05); in TEST_F()
2883 test_layer_norm({3, 11, 5, 7}, {5, 7}, {5, 7}, {5, 7}, 1e-05); in TEST_F()
2884 test_layer_norm({3, 11, 5, 7}, {11, 5, 7}, {11, 5, 7}, {11, 5, 7}, 1e-05); in TEST_F()
2885 test_layer_norm( in TEST_F()
|
| /external/pytorch/test/ |
| D | test_jit_llga_fuser.py | 341 def test_layer_norm(self, dtype): member in TestOp
|
| D | test_nestedtensor.py | 930 def test_layer_norm(self, device, dtype): member in TestNestedTensorDeviceType
|
| D | test_mps.py | 2680 def test_layer_norm(self): member in TestMPS
|
| /external/pytorch/test/inductor/ |
| D | test_perf.py | 416 def test_layer_norm(self): member in FusionTests
|
| D | test_torchinductor.py | 4546 def test_layer_norm(self): member in CommonTemplate
|
| /external/pytorch/test/quantization/core/ |
| D | test_quantized_module.py | 1052 def test_layer_norm(self): member in TestStaticQuantizedModule
|
| /external/pytorch/test/fx/ |
| D | test_z3_gradual_types.py | 392 def test_layer_norm(self): member in HFOperations
|
| /external/pytorch/test/quantization/jit/ |
| D | test_quantize_jit.py | 2677 def test_layer_norm(self): member in TestQuantizeJitOps
|
| /external/pytorch/test/quantization/fx/ |
| D | test_quantize_fx.py | 7731 def test_layer_norm(self): member in TestQuantizeFxOps
|