• Home
  • Raw
  • Download

Lines Matching defs:ConvParams

285 struct ConvParams {  struct
286 std::vector<T> stride;
287 std::vector<T> padding;
288 std::vector<T> dilation;
289 bool transposed{};
290 std::vector<T> output_padding;
291 T groups{};
292 bool benchmark{};
293 bool deterministic{};
294 bool cudnn_enabled{};
295 bool allow_tf32{};
297 bool is_strided() const { in is_strided()
305 bool is_dilated() const { in is_dilated()
313 bool is_padded() const { in is_padded()
321 bool is_output_padding_neg() const { in is_output_padding_neg()
329 bool is_output_padding_big() const { in is_output_padding_big()
337 bool is_padding_neg() const { in is_padding_neg()
345 bool is_dilation_neg() const { in is_dilation_neg()
353 bool is_stride_nonpos() const { in is_stride_nonpos()
361 void view1d_as_2d() { in view1d_as_2d()
370 …d(const at::Tensor& input, const at::Tensor& weight, const std::optional<at::Tensor>& bias) const { in use_cpu_depthwise3x3_winograd()
395 bool needs_64bit_indexing_no_split(const at::Tensor& input, const at::Tensor& weight) const { in needs_64bit_indexing_no_split()
419 bool use_cudnn(const at::Tensor& input, const at::Tensor& weight) const { in use_cudnn()
461 bool use_cudnn_depthwise(const at::Tensor& input, const at::Tensor& weight) const { in use_cudnn_depthwise()
504 bool use_miopen(const at::Tensor& input, const at::Tensor& weight, bool bias_defined) const { in use_miopen()
516 bool use_mkldnn(const at::Tensor& input, const at::Tensor& weight) const { in use_mkldnn()
545 bool use_nnpack(const at::Tensor& input, const at::Tensor& weight) const { in use_nnpack()
564 bool use_xnnpack(const at::Tensor& input, const at::Tensor& weight, in use_xnnpack()
586 bool use_mps(const at::Tensor& input, const at::Tensor& weight) const { in use_mps()
605 bool is_depthwise(const at::Tensor& input, const at::Tensor& weight) const { in is_depthwise()