Searched refs:softmax_script (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/benchmarks/static_runtime/ |
| H A D | test_static_runtime.cc | 2073 const auto softmax_script = R"JIT( in TEST() local
2086 testStaticRuntime(softmax_script, {a, 0}); in TEST()
2087 testStaticRuntime(softmax_script, {a, 1}); in TEST()
2089 testStaticRuntime(softmax_script, {b, 0}); in TEST()
2090 testStaticRuntime(softmax_script, {b, 1}); in TEST()
2091 testStaticRuntime(softmax_script, {b, 2}); in TEST()
|