Migrate concat layer

This commit is contained in:
2025-11-21 23:52:58 +01:00
parent fd4775faa4
commit aeb1739c46
9 changed files with 90 additions and 96 deletions

View File

@@ -80,6 +80,12 @@ class CUDA : public Backend {
CUDANet::Tensor& running_var,
CUDANet::Tensor& epsilon
) override;
CUDANet::Tensor& concat(
CUDANet::Tensor& input_a,
CUDANet::Tensor& input_b,
CUDANet::Tensor& output
) override;
};
} // namespace CUDANet::Backend