/* Copyright 2019 Google Inc. All Rights Reserved.

* Licensed under the Apache License, Version 2.0 (the "License");

* you may not use this file except in compliance with the License.

* You may obtain a copy of the License at

*

* http://www.apache.org/licenses/LICENSE-2.0

*

* Unless required by applicable law or agreed to in writing, software

* distributed under the License is distributed on an "AS IS" BASIS,

* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

* See the License for the specific language governing permissions and

* limitations under the License.

* ===========================================================================*/

#ifndef CONV2D_IMPL_H_

#define CONV2D_IMPL_H_

#include <cstddef>

namespace tfjs {

namespace wasm {

// Must match enum in FusedConv2D.ts.

enum FusableActivation { LINEAR = 0, RELU = 1, RELU6 = 2, PRELU = 3 };

void conv2d(const size_t x_id, const size_t batch_size,

const size_t input_height, const size_t input_width,

const size_t filter_id, const size_t filter_height,

const size_t filter_width, const size_t bias_id, size_t pad_top,

size_t pad_right, size_t pad_bottom, size_t pad_left,

const bool is_same_pad, const size_t dilation_height,

const size_t dilation_width, const size_t stride_height,

const size_t stride_width, const size_t input_channels,

const size_t output_channels, const bool is_depthwise,

const FusableActivation activation, const size_t prelu_weighs_id,

const size_t out_id);

} // namespace wasm

} // namespace tfjs

#endif // CONV2D_IMPL_H_