Esempio n. 1
0
        Tensor IOps.DepthToSpace(Tensor X, int[] scale, Layer.DepthToSpaceMode mode)
        {
            var O = m_Ops.DepthToSpace(X, scale, mode);

            m_Mem += (long)X.length + (long)O.length;
            return(O);
        }
Esempio n. 2
0
 /// <inheritdoc/>
 Tensor IOps.DepthToSpace(Tensor X, int[] scale, Layer.DepthToSpaceMode mode)
 {
     var O = m_Ops.DepthToSpace(X, scale, mode);
     LogLayerSummary(X.shape + " ^ " + mode + O.shape);
     LogOutputTensorSummary(O, Prefix + "DepthToSpace");
     return O;
 }
Esempio n. 3
0
        /// <inheritdoc/>
        Tensor IOps.DepthToSpace(Tensor X, int[] scale, Layer.DepthToSpaceMode mode)
        {
            var Y = m_Ops1.DepthToSpace(X, scale, mode);
            var Z = m_Ops2.DepthToSpace(X, scale, mode);

            CheckSame(Y, Z, Layer.Type.DepthToSpace);
            return(Y);
        }
Esempio n. 4
0
        /// <inheritdoc/>
        Tensor IOps.DepthToSpace(Tensor X, int[] scale, Layer.DepthToSpaceMode mode)
        {
            var O = m_Ops.DepthToSpace(X, scale, mode);

            D.Log(X.shape + " ^ " + mode + O.shape);
            O.PrintDataPart(32, Prefix + "DepthToSpace");
            return(O);
        }