Method get_kernel_dims

src/layers/learning/convolution.cpp:118–126

template <typename TensorDataType, data_layout Layout, El::Device Device>
std::vector<int>
convolution_layer<TensorDataType, Layout, Device>::get_kernel_dims() const
{
  std::vector<int> dims;
  dims.push_back(this->m_output_channels);
  dims.push_back(this->get_input_dims()[0] / this->m_groups);
  dims.insert(dims.end(), this->m_conv_dims.begin(), this->m_conv_dims.end());
  return dims;
}

template <typename TensorDataType, data_layout Layout, El::Device Device>
void convolution_layer<TensorDataType, Layout, Device>::fp_compute()