Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dpasses.cpp75 std::string fused_pattern = R"IR( in ConcatAddMulReplaceNaNClip() local
81 fuse.RegisterRewritePattern(pattern, fused_pattern); in ConcatAddMulReplaceNaNClip()
84 fuse.RegisterRewritePattern(pattern2, fused_pattern); in ConcatAddMulReplaceNaNClip()
87 fuse.RegisterRewritePattern(pattern3, fused_pattern); in ConcatAddMulReplaceNaNClip()
90 fuse.RegisterRewritePattern(pattern4, fused_pattern); in ConcatAddMulReplaceNaNClip()
103 std::string fused_pattern = R"IR( in CastedBatchOneHotLengths() local
108 fuse.RegisterRewritePattern(pattern, fused_pattern); in CastedBatchOneHotLengths()
135 std::string fused_pattern = R"IR( in ConcatBatchMatMulBatchGather() local
140 fuse.RegisterRewritePattern(pattern, fused_pattern); in ConcatBatchMatMulBatchGather()
183 std::string fused_pattern = R"IR( in ClipRangesGatherRangesLengthsToOffsets() local
[all …]