Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/cpp/jit/
H A Dtest_stack_opt.cpp13 TEST(StackOptTest, UseVariadicStack) { in TEST() argument
39 ASSERT_TRUE(UseVariadicStack(graph)); in TEST()
84 ASSERT_TRUE(UseVariadicStack(graph)); in TEST()
124 ASSERT_TRUE(UseVariadicStack(graph)); in TEST()
167 ASSERT_TRUE(UseVariadicStack(graph)); in TEST()
213 ASSERT_FALSE(UseVariadicStack(graph)); in TEST()
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dvariadic_ops.h25 TORCH_API bool UseVariadicStack(const std::shared_ptr<Graph>& graph);
H A Dvariadic_ops.cpp227 bool UseVariadicStack(const std::shared_ptr<Graph>& graph) { in UseVariadicStack() function
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dimpl.cpp179 UseVariadicStack(graph); in OptimizeGraph()