Searched defs:GpuTensorInfo (Results 1 – 3 of 3) sorted by relevance
107 struct GpuTensorInfo { struct109 …explicit GpuTensorInfo(const MSTensor *tensor, registry::opencl::OpenCLRuntimeWrapper *opencl_run)… in GpuTensorInfo() argument142 size_t N{1};143 size_t H{1};144 size_t W{1};145 size_t C{1};146 size_t Slice{};147 size_t width{};148 size_t height{};149 size_t FLT_size{4};[all …]
74 struct GpuTensorInfo { struct76 …explicit GpuTensorInfo(const MSTensor *tensor, registry::opencl::OpenCLRuntimeWrapper *opencl_run)… in GpuTensorInfo() argument
92 struct GpuTensorInfo { struct94 explicit GpuTensorInfo(const lite::Tensor *tensor) { in GpuTensorInfo() function133 size_t RowPitch() const { in RowPitch()141 int AlignAxis(int oriAxis) const { in AlignAxis()152 bool IsImageSizeValid() { return width > 0 && height > 0; } in IsImageSizeValid()