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: