Searched defs:SpatialTransformerDescriptor (Results 1 – 1 of 1) sorted by relevance
217 struct TORCH_CUDA_CPP_API SpatialTransformerDescriptor struct222 void set(cudnnDataType_t dataType, int dim, int* size) { in set()