def apply(self, input, mode): out_x = ops.conv_3d(input, self._w_x, self._pad_mode) out_y = ops.conv_3d(input, self._w_y, self._pad_mode) out_z = ops.conv_3d(input, self._w_z, self._pad_mode) # concatenate together. out = self._crop_sub_outputs_same_dims_and_concat(out_x, out_y, out_z) return out
def apply(self, input, mode): return ops.conv_3d(input, self._w, self._pad_mode)