(self, inp, filter)
| 1158 | name=self.name) |
| 1159 | |
| 1160 | def __call__(self, inp, filter): # pylint: disable=redefined-builtin |
| 1161 | # copybara:strip_begin |
| 1162 | # TODO(b/138808492): Remove code inside copybara |
| 1163 | # to make TPU code and CPU code consistent. |
| 1164 | # TPU convolution supports dilations greater than 1. |
| 1165 | if _enclosing_tpu_context() is not None: |
| 1166 | return convolution_internal( |
| 1167 | inp, |
| 1168 | filter, |
| 1169 | strides=self.strides, |
| 1170 | padding=self.padding, |
| 1171 | data_format=self.data_format, |
| 1172 | dilations=self.dilation_rate, |
| 1173 | name=self.name, |
| 1174 | call_from_convolution=False) |
| 1175 | else: |
| 1176 | return self.conv_op(inp, filter) |
| 1177 | # copybara:strip_end |
| 1178 | # copybara:insert return self.conv_op(inp, filter) |
| 1179 | |
| 1180 | |
| 1181 | @tf_export(v1=["nn.pool"]) |
nothing calls this directly
no test coverage detected