Method run

src/gpu/cl/operators/ClPRelu.cpp:50–61

}

void ClPRelu::run(ITensorPack &tensors)
{
    // Output tensor can be given as nullptr for in-place computation.
    // In this case, get the input tensor and use it as the output tensor.
    if (tensors.get_tensor(TensorType::ACL_DST) == nullptr)
    {
        auto src_tensor = const_cast<ITensor *>(tensors.get_const_tensor(TensorType::ACL_SRC_0));
        ARM_COMPUTE_ERROR_ON_MSG(src_tensor == nullptr, "invalid source tensor is given for in-place computation");
        tensors.add_tensor(TensorType::ACL_DST, src_tensor);
    }
    IClOperator::run(tensors);
}
} // namespace opencl
} // namespace arm_compute