Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/benchmarks/static_runtime/
H A Dtest_utils.cc224 return smodule.findNodeWithKindForTesting(kind); in getNodeWithKind()
408 return smodule.findNodeWithKindForTesting(name) != nullptr; in hasProcessedNodeWithName()
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dimpl.h491 C10_NODISCARD Node* findNodeWithKindForTesting(const std::string& kind) const;
H A Dimpl.cpp875 Node* StaticModule::findNodeWithKindForTesting(const std::string& kind) const { in findNodeWithKindForTesting() function in torch::jit::StaticModule