Contains all information related to a found Fused/UnfusedBatchNorm.
| 1012 | |
| 1013 | |
| 1014 | class _BatchNormMatch(object): |
| 1015 | """Contains all information related to a found Fused/UnfusedBatchNorm.""" |
| 1016 | |
| 1017 | def __init__(self, layer_op, bn_op, output_tensor, input_tensor, |
| 1018 | weight_tensor, gamma_tensor, beta_tensor, mean_tensor, |
| 1019 | variance_tensor, moving_mean_tensor, moving_variance_tensor, |
| 1020 | bn_decay_mean_tensor, bn_decay_var_tensor, batch_epsilon, |
| 1021 | batch_to_space_op): |
| 1022 | self._layer_op = layer_op |
| 1023 | self._bn_op = bn_op |
| 1024 | self._output_tensor = output_tensor |
| 1025 | self._input_tensor = input_tensor |
| 1026 | self._weight_tensor = weight_tensor |
| 1027 | self._gamma_tensor = gamma_tensor |
| 1028 | self._beta_tensor = beta_tensor |
| 1029 | self._mean_tensor = mean_tensor |
| 1030 | self._variance_tensor = variance_tensor |
| 1031 | self._moving_mean_tensor = moving_mean_tensor |
| 1032 | self._moving_variance_tensor = moving_variance_tensor |
| 1033 | self._bn_decay_mean_tensor = bn_decay_mean_tensor |
| 1034 | self._bn_decay_var_tensor = bn_decay_var_tensor |
| 1035 | self._batch_epsilon = batch_epsilon |
| 1036 | self._batch_to_space_op = batch_to_space_op |
| 1037 | |
| 1038 | @property |
| 1039 | def layer_op(self): |
| 1040 | return self._layer_op |
| 1041 | |
| 1042 | @property |
| 1043 | def bn_op(self): |
| 1044 | return self._bn_op |
| 1045 | |
| 1046 | @property |
| 1047 | def output_tensor(self): |
| 1048 | return self._output_tensor |
| 1049 | |
| 1050 | @property |
| 1051 | def input_tensor(self): |
| 1052 | return self._input_tensor |
| 1053 | |
| 1054 | @property |
| 1055 | def weight_tensor(self): |
| 1056 | return self._weight_tensor |
| 1057 | |
| 1058 | @property |
| 1059 | def gamma_tensor(self): |
| 1060 | return self._gamma_tensor |
| 1061 | |
| 1062 | @property |
| 1063 | def beta_tensor(self): |
| 1064 | return self._beta_tensor |
| 1065 | |
| 1066 | @property |
| 1067 | def mean_tensor(self): |
| 1068 | return self._mean_tensor |
| 1069 | |
| 1070 | @property |
| 1071 | def variance_tensor(self): |
no outgoing calls
no test coverage detected