Line data    Source code

       1             : #include "Input.h"
       2             : 
       3             : namespace elsa::ml
       4             : {
       5             :     template <typename data_t>
       6           4 :     Input<data_t>::Input(const VolumeDescriptor& inputDescriptor, index_t batchSize,
       7             :                          const std::string& name)
       8             :         : Layer<data_t>(LayerType::Input, name, Layer<data_t>::AnyNumberOfInputDimensions,
       9             :                         /* allowedNumberOfInputs */ 0),
      10           4 :           batchSize_(batchSize)
      11             :     {
      12           4 :         this->setInputDescriptor(inputDescriptor);
      13           4 :     }
      14             : 
      15             :     template <typename data_t>
      16           1 :     index_t Input<data_t>::getBatchSize() const
      17             :     {
      18           1 :         return batchSize_;
      19             :     }
      20             : 
      21             :     template class Input<float>;
      22             : } // namespace elsa::ml