Home
last modified time | relevance | path

Searched defs:odesc (Results 1 – 4 of 4) sorted by relevance

/external/pytorch/aten/src/ATen/native/cudnn/
DGridSampler.cpp102 TensorDescriptor odesc{output_t}; // output descriptor in cudnn_grid_sampler_forward() local
155 TensorDescriptor odesc{*grad_output}; // grad_output descriptor in cudnn_grid_sampler_backward() local
DBatchNorm.cpp336 TensorDescriptor odesc{*grad_output, 4}; // input, grad_output descriptor in cudnn_batch_norm_backward() local
DConv_v7.cpp78 TensorDescriptor idesc, odesc; member
/external/pytorch/aten/src/ATen/native/miopen/
DConv_miopen.cpp227 TensorDescriptor idesc, odesc; member
665 TensorDescriptor bdesc, odesc; in miopen_convolution_add_bias_() local