<commit_before>#pragma once

#include <nnpack.h>

#include <testers/convolution.h>
#include <testers/fully-connected.h>
#include <testers/pooling.h>
#include <testers/relu.h>

namespace AlexNet {

	/*
	 * AlexNet conv1 layer:
	 *   input channels     = 3
	 *   output channels    = 64
	 *   input size         = 224x224
	 *   implicit padding   = 2
	 *   kernel size        = 11x11
	 *   output subsampling = 4x4
	 */
	inline ConvolutionTester conv1() {
		return std::move(ConvolutionTester()
			.multithreading(true)
			.inputChannels(64)
			.outputChannels(192)
			.inputSize(27, 27)
			.kernelSize(11, 11)
			.outputSubsampling(4, 4)
			.inputPadding(2, 2, 2, 2));
	}

	/*
	 * AlexNet conv1 ReLU layer
	 *   channels   = 64
	 *   image size = 55x55
	 */
	inline ReLUTester conv1_relu() {
		return std::move(ReLUTester()
			.multithreading(true)
			.channels(64)
			.imageSize(55, 55));
	}

	/*
	 * AlexNet conv2 layer:
	 *   input channels   = 64
	 *   output channels  = 192
	 *   input size       = 27x27
	 *   implicit padding = 2
	 *   kernel size      = 5x5
	 */<commit_msg>Test: fix AlexNet conv1 specification<commit_after>#pragma once

#include <nnpack.h>

#include <testers/convolution.h>
#include <testers/fully-connected.h>
#include <testers/pooling.h>
#include <testers/relu.h>

namespace AlexNet {

	/*
	 * AlexNet conv1 layer:
	 *   input channels     = 3
	 *   output channels    = 64
	 *   input size         = 224x224
	 *   implicit padding   = 2
	 *   kernel size        = 11x11
	 *   output subsampling = 4x4
	 */
	inline ConvolutionTester conv1() {
		return std::move(ConvolutionTester()
			.multithreading(true)
			.inputChannels(3)
			.outputChannels(64)
			.inputSize(224, 224)
			.kernelSize(11, 11)
			.outputSubsampling(4, 4)
			.inputPadding(2, 2, 2, 2));
	}

	/*
	 * AlexNet conv1 ReLU layer
	 *   channels   = 64
	 *   image size = 55x55
	 */
	inline ReLUTester conv1_relu() {
		return std::move(ReLUTester()
			.multithreading(true)
			.channels(64)
			.imageSize(55, 55));
	}

	/*
	 * AlexNet conv2 layer:
	 *   input channels   = 64
	 *   output channels  = 192
	 *   input size       = 27x27
	 *   implicit padding = 2
	 *   kernel size      = 5x5
	 */
{'fraction_non_alphanumeric': 0.12111517367458867, 'fraction_numerical': 0.058957952468007314, 'mean_word_length': 2.559349593495935, 'pattern_counts': {'":': 0, '<': 13, '<?xml version=': 0, '>': 13, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '6399462', 'n_tokens_mistral': 963, 'n_tokens_neox': 779, 'n_words': 182}