Searched refs:mGpuReluParam (Results 1 – 2 of 2) sorted by relevance
22 …mGpuReluParam.reset(new VulkanBuffer(vulkanBn->getMemoryPool(), false, sizeof(GpuReluParam), nullp… in VulkanRelu()65 auto reluParam = reinterpret_cast<GpuReluParam *>(mGpuReluParam->map()); in onEncode()74 mGpuReluParam->unmap(); in onEncode()80 mDescriptorSet->writeBuffer(mGpuReluParam->buffer(), 2, mGpuReluParam->size()); in onEncode()
27 std::shared_ptr<VulkanBuffer> mGpuReluParam; member in MNN::VulkanRelu