org.apache.sysml.runtime.matrix.data

Class ConvolutionParameters

  • All Implemented Interfaces:
    Serializable


    public class ConvolutionParameters
    extends Object
    implements Serializable
    This class is container that stores parameters required for executing following operations: conv2d, conv2d_backward_data, conv2d_backward_filter, maxpooling, maxpooling_backward
    See Also:
    Serialized Form
    • Field Detail

      • N

        public int N
      • C

        public int C
      • H

        public int H
      • W

        public int W
      • K

        public int K
      • R

        public int R
      • S

        public int S
      • stride_h

        public int stride_h
      • stride_w

        public int stride_w
      • pad_h

        public int pad_h
      • pad_w

        public int pad_w
      • P

        public int P
      • Q

        public int Q
      • numThreads

        public int numThreads
      • enableNative

        public boolean enableNative
      • start_indexes_h

        public int[] start_indexes_h
      • end_indexes_h

        public int[] end_indexes_h
      • start_indexes_w

        public int[] start_indexes_w
      • end_indexes_w

        public int[] end_indexes_w
    • Constructor Detail

      • ConvolutionParameters

        public ConvolutionParameters(long N,
                                     long C,
                                     long H,
                                     long W,
                                     long K,
                                     long R,
                                     long S,
                                     long stride_h,
                                     long stride_w,
                                     long pad_h,
                                     long pad_w,
                                     int numThreads)
                              throws DMLRuntimeException
        Throws:
        DMLRuntimeException
      • ConvolutionParameters

        public ConvolutionParameters(int N,
                                     int C,
                                     int H,
                                     int W,
                                     int K,
                                     int R,
                                     int S,
                                     int stride_h,
                                     int stride_w,
                                     int pad_h,
                                     int pad_w,
                                     int numThreads)

Copyright © 2017 The Apache Software Foundation. All rights reserved.