device_grouped_conv_fwd_multiple_d.hpp Source File

• include
• ck
• tensor_operation
• gpu
• device

device_grouped_conv_fwd_multiple_d.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 "ck/tensor_operation/gpu/device/device_grouped_conv_fwd_multiple_abd.hpp"

7#include "ck/tensor_operation/gpu/device/impl/device_grouped_conv_utils.hpp"

8

9namespace ck {

10namespace tensor_operation {

11namespace device {

12

33template <index_t NDimSpatial,

34 typename ALayout,

35 typename BLayout,

36 typename DsLayout,

37 typename ELayout,

38 typename ADataType,

39 typename BDataType,

40 typename DsDataType,

41 typename EDataType,

42 typename AElementwiseOperation,

43 typename BElementwiseOperation,

44 typename CDEElementwiseOperation,

45 typename ComputeType =

46 decltype(UnpackDataType<is_detected<is_tuple, ADataType>::value,

47 Number<0>,

48 ADataType>())> // ComputeType is InputType by default (first

49 // in tuple for MultiAB), unpack if tuple was

50 // passed

51using DeviceGroupedConvFwdMultipleD = DeviceGroupedConvFwdMultipleABD<NDimSpatial,

52 ALayout,

53 BLayout,

54 DsLayout,

55 ELayout,

56 ADataType,

57 BDataType,

58 DsDataType,

59 EDataType,

60 AElementwiseOperation,

61 BElementwiseOperation,

62 CDEElementwiseOperation,

63 ComputeType>;

64

65} // namespace device

66} // namespace tensor_operation

67} // namespace ck

device_grouped_conv_fwd_multiple_abd.hpp

device_grouped_conv_utils.hpp

ck::tensor_operation::device

Definition convolution_backward_data_specialization.hpp:8

ck::tensor_operation::device::DeviceGroupedConvFwdMultipleD

DeviceGroupedConvFwdMultipleABD< NDimSpatial, ALayout, BLayout, DsLayout, ELayout, ADataType, BDataType, DsDataType, EDataType, AElementwiseOperation, BElementwiseOperation, CDEElementwiseOperation, ComputeType > DeviceGroupedConvFwdMultipleD

Grouped Convolution Forward.

Definition device_grouped_conv_fwd_multiple_d.hpp:51

ck::tensor_operation

Definition convolution_backward_data_specialization.hpp:7

ck

Definition ck.hpp:268

ck::index_t

int32_t index_t

Definition ck.hpp:299

ck::Number

integral_constant< index_t, N > Number

Definition number.hpp:12

ck::tensor_operation::device::DeviceGroupedConvFwdMultipleABD

Grouped Convolution Forward.

Definition device_grouped_conv_fwd_multiple_abd.hpp:73