Home
last modified time | relevance | path

Searched refs:findChildModule (Results 1 – 5 of 5) sorted by relevance

/external/pytorch/torch/csrc/jit/passes/quantization/
Ddedup_module_uses.cpp55 auto m = findChildModule(module_, path); in findModuleUses()
71 const auto& m = findChildModule(module_, path); in dedupModuleUses()
94 auto parent_of_leaf = findChildModule( in addChildModule()
Dhelper.h144 findChildModule(const Module& module, const std::vector<std::string>& path);
Dhelper.cpp629 Module findChildModule( in findChildModule() function
642 return findChildModule(module, path); in getInvokedModule()
/external/pytorch/torch/csrc/jit/passes/
Dfold_conv_bn.cpp303 Module conv_submodule = findChildModule(current, conv_module_path); in analyze()
304 Module bn_submodule = findChildModule(current, bn_module_path); in analyze()
340 Module conv_submodule = findChildModule(current, std::get<0>(conv_bn)); in analyze()
341 Module bn_submodule = findChildModule(current, std::get<1>(conv_bn)); in analyze()
Dhoist_conv_packed_params.cpp53 Module convModule = findChildModule(rootModule, rootToConvPath); in hoistConvPackedParams()