Home
last modified time | relevance | path

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

/external/pytorch/test/cpp/jit/
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()
/external/pytorch/torch/csrc/jit/passes/
Dvariadic_ops.h25 TORCH_API bool UseVariadicStack(const std::shared_ptr<Graph>& graph);
Dvariadic_ops.cpp227 bool UseVariadicStack(const std::shared_ptr<Graph>& graph) { in UseVariadicStack() function
/external/pytorch/torch/csrc/jit/runtime/static/
Dimpl.cpp179 UseVariadicStack(graph); in OptimizeGraph()