Searched defs:StaticRuntime (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/benchmarks/static_runtime/ |
| H A D | test_static_runtime.cc | 52 TEST(StaticRuntime, UnaryOps) { in TEST() argument
97 TEST(StaticRuntime, Max) { in TEST() argument
134 TEST(StaticRuntime, Mean) { in TEST() argument
180 TEST(StaticRuntime, Sigmoid) { in TEST() argument
200 TEST(StaticRuntime, Clone) { in TEST() argument
271 TEST(StaticRuntime, Clamp) { in TEST() argument
298 TEST(StaticRuntime, ClampMinOnly) { in TEST() argument
310 TEST(StaticRuntime, ClampMaxOnly) { in TEST() argument
322 TEST(StaticRuntime, ClampIntTensor) { in TEST() argument
336 TEST(StaticRuntime, LenWithTuple) { in TEST() argument
[all …]
|
| H A D | test_generated_ops.cc | 15 TEST(StaticRuntime, autogen_absolute) { in TEST() argument
45 TEST(StaticRuntime, autogen_angle) { in TEST() argument
75 TEST(StaticRuntime, autogen_sgn) { in TEST() argument
105 TEST(StaticRuntime, autogen_acos) { in TEST() argument
135 TEST(StaticRuntime, autogen_arccos) { in TEST() argument
165 TEST(StaticRuntime, autogen__add_relu_Tensor) { in TEST() argument
199 TEST(StaticRuntime, autogen_addmv) { in TEST() argument
237 TEST(StaticRuntime, autogen_addr) { in TEST() argument
275 TEST(StaticRuntime, autogen__test_functorch_fallback) { in TEST() argument
307 TEST(StaticRuntime, autogen_argmax) { in TEST() argument
[all …]
|
| H A D | test_static_module.cc | 213 TEST(StaticRuntime, ModuleHasOp) { in TEST() argument
220 TEST(StaticRuntime, ReplaceWithCopy_replaces_reshape) { in TEST() argument
256 StaticRuntime, in TEST() argument
301 TEST(StaticRuntime, CanEnableStaticRuntime) { in TEST() argument
356 TEST(StaticRuntime, CanEnableStaticRuntimeCallMethod) { in TEST() argument
365 TEST(StaticRuntime, CanEnableStaticRuntimeSubBlocks) { in TEST() argument
377 TEST(StaticRuntime, NestedOutput) { in TEST() argument
447 TEST(StaticRuntime, LongModel) { in TEST() argument
466 TEST(StaticRuntime, TrivialModel) { in TEST() argument
485 TEST(StaticRuntime, DeepWide) { in TEST() argument
[all …]
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/ |
| H A D | impl.h | 244 class StaticRuntime; variable
|
| H A D | impl.cpp | 2256 StaticRuntime::StaticRuntime(const StaticModule& sm) in StaticRuntime() function in torch::jit::StaticRuntime
|