Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/benchmarks/static_runtime/
H A Dtest_static_runtime.cc2073 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()