Line data    Source code

       1             : #include "DnnlNoopLayer.h"
       2             : 
       3             : namespace elsa::ml
       4             : {
       5             :     namespace detail
       6             :     {
       7             :         template <typename data_t>
       8           0 :         DnnlNoopLayer<data_t>::DnnlNoopLayer(const VolumeDescriptor& inputDescriptor)
       9           0 :             : DnnlLayer<data_t>(inputDescriptor, inputDescriptor, "DnnlNoopLayer")
      10             :         {
      11           0 :         }
      12             : 
      13             :         template <typename data_t>
      14           0 :         void DnnlNoopLayer<data_t>::compileForwardStream()
      15             :         {
      16           0 :             BaseType::compileForwardStream();
      17           0 :             BaseType::validateDnnlMemory(_input.front().effectiveMemory);
      18           0 :             _output.effectiveMemory = _input.front().effectiveMemory;
      19           0 :         }
      20             : 
      21             :         template <typename data_t>
      22           0 :         void DnnlNoopLayer<data_t>::compileBackwardStream()
      23             :         {
      24           0 :             BaseType::compileBackwardStream();
      25           0 :             BaseType::validateDnnlMemory(_outputGradient.front().effectiveMemory);
      26           0 :             _inputGradient.front().effectiveMemory = _outputGradient.front().effectiveMemory;
      27           0 :         }
      28             : 
      29             :         template class DnnlNoopLayer<float>;
      30             :     } // namespace detail
      31             : } // namespace elsa::ml