82    auto J = Reshape(j.Read(), NQ, 2, 2, NE);  in PAGradientSetup2D()
83    auto y = Reshape(op.Write(), NQ, 2, 2, NE);  in PAGradientSetup2D()
86    const auto C = const_c ? Reshape(c.Read(), 1,1) :  in PAGradientSetup2D()
87                   Reshape(c.Read(), NQ, NE);  in PAGradientSetup2D()
117    auto J = Reshape(j.Read(), NQ, 3, 3, NE);  in PAGradientSetup3D()
118    auto y = Reshape(op.Write(), NQ, 3, 3, NE);  in PAGradientSetup3D()
121    const auto C = const_c ? Reshape(c.Read(), 1,1) :  in PAGradientSetup3D()
122                   Reshape(c.Read(), NQ,NE);  in PAGradientSetup3D()
240       auto C = Reshape(coeff.HostWrite(), nq, ne);  in AssemblePA()
274    auto B = Reshape(b.Read(), Q1D, TR_D1D);  in PAGradientApply2D()
275    auto G = Reshape(g.Read(), Q1D, TR_D1D);  in PAGradientApply2D()
276    auto Bt = Reshape(bt.Read(), TE_D1D, Q1D);  in PAGradientApply2D()
277    auto op = Reshape(op_.Read(), Q1D*Q1D, 2,2, NE);  in PAGradientApply2D()
278    auto x = Reshape(x_.Read(), TR_D1D, TR_D1D, NE);  in PAGradientApply2D()
279    auto y = Reshape(y_.ReadWrite(), TE_D1D, TE_D1D, 2, NE);  in PAGradientApply2D()
404    auto B = Reshape(b.Read(), Q1D, TR_D1D);  in PAGradientApply3D()
405    auto G = Reshape(g.Read(), Q1D, TR_D1D);  in PAGradientApply3D()
406    auto Bt = Reshape(bt.Read(), TE_D1D, Q1D);  in PAGradientApply3D()
407    auto op = Reshape(op_.Read(), Q1D*Q1D*Q1D, 3,3, NE);  in PAGradientApply3D()
408    auto x = Reshape(x_.Read(), TR_D1D, TR_D1D, TR_D1D, NE);  in PAGradientApply3D()
409    auto y = Reshape(y_.ReadWrite(), TE_D1D, TE_D1D, TE_D1D, 3, NE);  in PAGradientApply3D()
603    auto b = Reshape(b_.Read(), Q1D, TR_D1D);  in SmemPAGradientApply3D()
604    auto g = Reshape(g_.Read(), Q1D, TR_D1D);  in SmemPAGradientApply3D()
605    auto bt = Reshape(bt_.Read(), TE_D1D, Q1D);  in SmemPAGradientApply3D()
606    auto D = Reshape(d_.Read(), Q1D*Q1D*Q1D, 3, 3, NE);  in SmemPAGradientApply3D()
607    auto x = Reshape(x_.Read(), TR_D1D, TR_D1D, TR_D1D, NE);  in SmemPAGradientApply3D()
608    auto y = Reshape(y_.ReadWrite(), TE_D1D, TE_D1D, TE_D1D, 3, NE);  in SmemPAGradientApply3D()