Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/metal/mpscnn/tests/
H A DMPSCNNTests.h39 bool test_log_softmax();
H A DMPSCNNTests.mm827 bool test_log_softmax() { function
/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_pytorch_jit_onnx.py152 def test_log_softmax(self): member in _TestJITIRToONNX
/aosp_15_r20/external/pytorch/test/
H A Dtest_nnapi.py401 def test_log_softmax(self): member in TestNNAPI
H A Dtest_tensorexpr.py1270 def test_log_softmax(x, y): function
1277 for test in (test_softmax, test_log_softmax, test_softmax_neg_index):
H A Dtest_mps.py4999 def test_log_softmax(self): member in TestMPS
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/tests/
H A Dtf-to-tosa-pipeline.mlir641 // CHECK-LABEL: test_log_softmax
647 func.func @test_log_softmax(%arg0: tensor<13x21x3xf32>) -> tensor<13x21x3xf32> {
H A Dtfl-to-tosa-pipeline.mlir991 // CHECK-LABEL: test_log_softmax
997 func.func @test_log_softmax(%arg0: tensor<13x21x3xf32>) -> tensor<13x21x3xf32> {
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/
H A Dcontrol_flow_ops_test.py762 def test_log_softmax(self): member in NNTest
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_eager_transforms.py1763 def test_log_softmax(self, device): member in TestVmapOfGrad
H A Dtest_vmap.py3728 def test_log_softmax(self, device): member in TestVmapBatchedGradient
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_torchinductor.py2710 def test_log_softmax(self): member in CommonTemplate