Searched refs:_fused_can_be_used (Results 1 – 1 of 1) sorted by relevance
218 elif fused is None and not self._fused_can_be_used():273 def _fused_can_be_used(self): member in BatchNormalizationBase352 self.fused = (ndims == 4 and self._fused_can_be_used())1135 def _fused_can_be_used(self, ndims): member in LayerNormalization1202 self._fused = self._fused_can_be_used(ndims)