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)