Searched refs:set_fuser_method (Results 1 – 8 of 8) sorted by relevance
196 .set_fuser_method(_fuse_conv_bn_add_left)209 .set_fuser_method(_fuse_conv_add_left)282 .set_fuser_method(_fuse_conv_bn_add_right)295 .set_fuser_method(_fuse_conv_add_right)388 .set_fuser_method(_fuse_conv_bn_add_relu_left)401 .set_fuser_method(_fuse_conv_add_relu_left)484 .set_fuser_method(_fuse_conv_bn_add_relu_right)497 .set_fuser_method(_fuse_conv_add_relu_right)538 .set_fuser_method(fuser_method)545 .set_fuser_method(fuser_method)[all …]
216 .set_fuser_method(_sequential_wrapper2(nni.LinearReLU))223 .set_fuser_method(_sequential_wrapper2(nni.LinearReLU))265 .set_fuser_method(fuse_linear_bn)331 .set_fuser_method(_sequential_wrapper2(convs.fused_conv_relu))338 .set_fuser_method(_sequential_wrapper2(convs.fused_conv_relu))394 .set_fuser_method(fuse_conv_bn)401 .set_fuser_method(fuse_conv_bn_relu)409 .set_fuser_method(fuse_conv_bn_relu)459 .set_fuser_method(fuse_convtranspose_bn)699 .set_fuser_method(_sequential_wrapper2(fused_bn))[all …]
185 .set_fuser_method(_sequential_wrapper2(convs.fused_conv_relu))192 .set_fuser_method(_sequential_wrapper2(convs.fused_conv_relu))244 .set_fuser_method(fuse_conv_bn)251 .set_fuser_method(fuse_conv_bn_relu)259 .set_fuser_method(fuse_conv_bn_relu)
94 .set_fuser_method(fuse_conv2d_relu)128 * `set_fuser_method`133 …ed during the prepare phase according to the function specified in `set_fuser_method`, which repla…
576 def set_fuser_method(self, fuser_method: Callable) -> BackendPatternConfig: member in BackendPatternConfig683 conf.set_fuser_method(
132 .set_fuser_method(self._fuser_method)207 conf.set_fuser_method(self._fuser_method)
108 .set_fuser_method(fuse_linear_relu)117 `set_fuser_method`: specifies the fuser method to use for the pattern, a fuser method will take the…
753 .set_fuser_method(fuse_conv_bn_relu)756 .set_fuser_method(fuse_conv_bn_relu)815 .set_fuser_method(fuse_conv_bn_relu) \878 .set_fuser_method(fuse_conv_relu)881 .set_fuser_method(fuse_conv_res_relu) \