Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dxnnpack_rewrite.cpp70 void transformConv1dToConv2d(std::shared_ptr<Graph>& graph) { in transformConv1dToConv2d() function
76 void transformConv1dToConv2d(script::Module& module) { in transformConv1dToConv2d() function
79 transformConv1dToConv2d(graph); in transformConv1dToConv2d()
82 transformConv1dToConv2d(m); in transformConv1dToConv2d()
426 transformConv1dToConv2d(cloned_module); in optimizeForMobile()
H A Dxnnpack_rewrite.h9 TORCH_API void transformConv1dToConv2d(std::shared_ptr<Graph>& graph);
10 TORCH_API void transformConv1dToConv2d(script::Module& module);
/aosp_15_r20/external/pytorch/aten/src/ATen/benchmarks/
H A Dstateful_conv1d.cpp43 torch::jit::transformConv1dToConv2d(m_cloned); in stateful_conv1d()
/aosp_15_r20/external/pytorch/torch/csrc/jit/python/
H A Dinit.cpp1041 return transformConv1dToConv2d(graph); in initJITBindings()
1046 return transformConv1dToConv2d(module); in initJITBindings()