mirror of
https://github.com/ml-explore/mlx.git
synced 2025-09-04 06:44:40 +08:00
More unused backend apis
This commit is contained in:
@@ -151,13 +151,6 @@ class Convolution {
|
|||||||
}
|
}
|
||||||
|
|
||||||
cudnnHandle_t handle_;
|
cudnnHandle_t handle_;
|
||||||
cudnnTensorDescriptor_t input_desc_{nullptr};
|
|
||||||
cudnnFilterDescriptor_t filter_desc_{nullptr};
|
|
||||||
cudnnTensorDescriptor_t output_desc_{nullptr};
|
|
||||||
cudnnConvolutionDescriptor_t conv_desc_{nullptr};
|
|
||||||
cudnnConvolutionFwdAlgo_t algo_{
|
|
||||||
CUDNN_CONVOLUTION_FWD_ALGO_IMPLICIT_PRECOMP_GEMM};
|
|
||||||
|
|
||||||
graph::Graph graph_;
|
graph::Graph graph_;
|
||||||
std::shared_ptr<graph::Tensor_attributes> input_attr_;
|
std::shared_ptr<graph::Tensor_attributes> input_attr_;
|
||||||
std::shared_ptr<graph::Tensor_attributes> filter_attr_;
|
std::shared_ptr<graph::Tensor_attributes> filter_attr_;
|
||||||
|
Reference in New Issue
Block a user