Method custom_norm

software/custom/custom_class.cpp:124–171

}

void Custom::custom_norm(const xChangeLayer *xchange){



	float *In_ptr;

	float *out_ptr;

	vector<int> out_dim;
	vector<int> in_dim;
	int batch ;
	int channels;
	int height;
	int width;

	int iN=0,oN=0;
	int off = 0;

	int local_size = (int)xchange->float_args[off+0];
	float alpha = xchange->float_args[off+1];
	float beta =xchange->float_args[off+2];
	for(int it_num_in=0;it_num_in<xchange->input_dims.size();it_num_in++){

		in_dim=xchange->input_dims[it_num_in];

		channels = in_dim[1];
		height = in_dim[2];
		width = in_dim[3];
		iN = channels*height*width;

	}

	for(int it_num_out=0;it_num_out<xchange->output_dims.size();it_num_out++){

		out_dim=xchange->output_dims[it_num_out];
		channels = out_dim[1];
		height = out_dim[2];
		width = out_dim[3];
		oN = channels*height*width;

	}
	batch = in_dim[0]; // default batch size = 2
	for(int batch_num=0;batch_num<batch;batch_num++){
		In_ptr = (float*)xchange->in_ptrs[0]+batch_num*iN;
		out_ptr = (float*)xchange->out_ptrs[0]+batch_num*oN;
		LRN_AcrossChannel(In_ptr, out_ptr, out_dim, local_size, alpha, beta);
	}
}

void func_memcopy(const float *In_ptr,float *out_ptr,int in_height,int in_width,int out_height,int out_width,int depth){