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()
|