/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-6.4.3/include/ck/tensor_operation/gpu/device/device_conv_bwd_data.hpp Source File

/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-6.4.3/include/ck/tensor_operation/gpu/device/device_conv_bwd_data.hpp Source File#

Composable Kernel: /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-6.4.3/include/ck/tensor_operation/gpu/device/device_conv_bwd_data.hpp Source File
device_conv_bwd_data.hpp
Go to the documentation of this file.
1 // SPDX-License-Identifier: MIT
2 // Copyright (c) 2018-2023, Advanced Micro Devices, Inc. All rights reserved.
3 
4 #pragma once
5 
6 #include <vector>
7 
8 #include "ck/tensor_operation/gpu/device/device_base.hpp"
9 
10 namespace ck {
11 namespace tensor_operation {
12 namespace device {
13 
14 template <ck::index_t NumDimSpatial,
15  typename InLayout,
16  typename WeiLayout,
17  typename OutLayout,
18  typename InDataType,
19  typename WeiDataType,
20  typename OutDataType,
21  typename InElementwiseOperation,
22  typename WeiElementwiseOperation,
23  typename OutElementwiseOperation>
24 struct DeviceConvBwdData : public BaseOperator
25 {
26  virtual std::unique_ptr<BaseArgument>
27  MakeArgumentPointer(void* p_in,
28  const void* p_wei,
29  const void* p_out,
30  ck::index_t N,
31  ck::index_t K,
32  ck::index_t C,
33  std::vector<ck::index_t> input_spatial_lengths,
34  std::vector<ck::index_t> filter_spatial_lengths,
35  std::vector<ck::index_t> output_spatial_lengths,
36  std::vector<ck::index_t> conv_filter_strides,
37  std::vector<ck::index_t> conv_filter_dilations,
38  std::vector<ck::index_t> input_left_pads,
39  std::vector<ck::index_t> input_right_pads,
40  InElementwiseOperation in_element_op,
41  WeiElementwiseOperation wei_element_op,
42  OutElementwiseOperation out_element_op) = 0;
43 
44  virtual std::unique_ptr<BaseInvoker> MakeInvokerPointer() = 0;
45 };
46 
47 } // namespace device
48 } // namespace tensor_operation
49 } // namespace ck
device_base.hpp
ck
Definition: ck.hpp:264
ck::index_t
int32_t index_t
Definition: ck.hpp:289
ck::tensor_operation::device::BaseOperator
Definition: device_base.hpp:76
ck::tensor_operation::device::DeviceConvBwdData
Definition: device_conv_bwd_data.hpp:25
ck::tensor_operation::device::DeviceConvBwdData::MakeArgumentPointer
virtual std::unique_ptr< BaseArgument > MakeArgumentPointer(void *p_in, const void *p_wei, const void *p_out, ck::index_t N, ck::index_t K, ck::index_t C, std::vector< ck::index_t > input_spatial_lengths, std::vector< ck::index_t > filter_spatial_lengths, std::vector< ck::index_t > output_spatial_lengths, std::vector< ck::index_t > conv_filter_strides, std::vector< ck::index_t > conv_filter_dilations, std::vector< ck::index_t > input_left_pads, std::vector< ck::index_t > input_right_pads, InElementwiseOperation in_element_op, WeiElementwiseOperation wei_element_op, OutElementwiseOperation out_element_op)=0
ck::tensor_operation::device::DeviceConvBwdData::MakeInvokerPointer
virtual std::unique_ptr< BaseInvoker > MakeInvokerPointer()=0