Searched defs:computation_call (Results 1 – 2 of 2) sorted by relevance
93 def _gelu_fusion_1(computation_call): argument107 def _gelu_fusion_2(computation_call): argument139 def _hardswish_fusion(computation_call): argument156 def _silu_fusion(computation_call): argument161 def _hardsigmoid_fusion(computation_call): argument174 def _leaky_relu_fusion(computation_call): argument182 def _hardtanh_fusion(computation_call): argument189 def _combined_fusion(computation_call, elementwise_op): argument193 def _binary_fusion_v1(computation_call, binary_fn): argument197 def _binary_fusion_v2(computation_call, binary_fn): argument
228 computation_call, argument253 def generate_pattern_with_unary(computation_call, unary_post_op): argument262 def generate_pattern_with_output_quant(computation_call, with_dtype_convert=False): argument