Home
last modified time | relevance | path

Searched refs:replaceConvolutionWithAtenConv (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dxnnpack_rewrite.cpp72 graph_rewrite_helper::replaceConvolutionWithAtenConv(graph); in transformConv1dToConv2d()
119 graph_rewrite_helper::replaceConvolutionWithAtenConv(graph); in insertPrePackedConv2dOp()
H A Dvulkan_rewrite.cpp78 graph_rewrite_helper::replaceConvolutionWithAtenConv(graph); in insertPrePackedConv2dOp()
121 graph_rewrite_helper::replaceConvolutionWithAtenConv(graph); in insertPrePackedConv1dOp()
H A Dgraph_rewrite_helper.h19 TORCH_API void replaceConvolutionWithAtenConv(std::shared_ptr<Graph>& graph);
H A Dfrozen_conv_add_relu_fusion_cuda.cpp119 graph_rewrite_helper::replaceConvolutionWithAtenConv(graph); in fuseFrozenConvAddReluImpl()
H A Dmetal_rewrite.cpp44 graph_rewrite_helper::replaceConvolutionWithAtenConv(graph); in insertPrePackedConv2dOp()
H A Dgraph_rewrite_helper.cpp54 void replaceConvolutionWithAtenConv(std::shared_ptr<Graph>& graph) { in replaceConvolutionWithAtenConv() function
H A Dfrozen_ops_to_mkldnn.cpp1140 graph_rewrite_helper::replaceConvolutionWithAtenConv(graph); in ConvertFrozenOpsToMKLDNN()
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/
H A Dinsert_observers.cpp44 using graph_rewrite_helper::replaceConvolutionWithAtenConv;
1196 replaceConvolutionWithAtenConv(graph); in preprocess()
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dkernel.cpp1664 graph_rewrite_helper::replaceConvolutionWithAtenConv(graph_); in optimizeOwningGraph()