Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Data Fields
at::native::ConvParams Struct Reference

Public Member Functions

bool is_strided () const
bool is_dilated () const
bool is_padded () const
bool is_output_padding_neg () const
bool is_output_padding_big () const
bool is_padding_neg () const
bool is_stride_neg () const
void view1d_as_2d ()
bool use_cudnn (const at::Tensor &input) const
bool use_miopen (const at::Tensor &input) const
bool use_mkldnn (const at::Tensor &input) const
bool use_nnpack (const at::Tensor &input) const
bool is_depthwise (const at::Tensor &input, const at::Tensor &weight) const

Data Fields

std::vector< int64_t > stride
std::vector< int64_t > padding
std::vector< int64_t > dilation
bool transposed
std::vector< int64_t > output_padding
int groups
bool benchmark
bool deterministic
bool cudnn_enabled

Detailed Description

Definition at line 13 of file Convolution.cpp.

The documentation for this struct was generated from the following file: