Searched defs:pW (Results 1 – 8 of 8) sorted by relevance
| /external/pytorch/aten/src/ATen/native/ |
| D | vol2col.h | 22 const int64_t pW, in vol2col()
73 const int64_t pW, in col2vol()
|
| D | DilatedMaxPool3d.cpp | 50 const int pW = padding.size() == 1 ? pT : safe_downcast<int, int64_t>(padding[2]); in max_pool3d_with_indices_out_cpu_template() local
142 const int pW = padding.size() == 1 ? pT : safe_downcast<int, int64_t>(padding[2]); in max_pool3d_with_indices_backward_out_cpu_template() local
|
| D | Pool.h | 293 int pT, int pH, int pW, in max_pool3d_backward_shape_check()
329 int pT, int pH, int pW, in avg_pool3d_backward_shape_check()
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| D | Pooling.cpp | 57 int64_t pW, // padding in spatial_dilated_max_pooling()
120 int64_t pW, // padding in spatial_dilated_max_pooling3d()
194 int64_t pW, // padding in q_maxpool_2d()
361 int64_t pW, // padding in q_maxpool_3d()
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | DilatedMaxPool3d.cu | 43 int pT, int pH, int pW, in max_pool3d_with_indices_single_out_frame()
142 int pT, int pH, int pW, in max_pool3d_with_indices_out_frame()
319 const int pW = padding.size() == 1 ? pT : safe_downcast<int, int64_t>(padding[2]); in max_pool3d_with_indices_out_cuda_template() local
449 const int pW = padding.size() == 1 ? pT : safe_downcast<int, int64_t>(padding[2]); in max_pool3d_with_indices_backward_out_cuda_template() local
|
| /external/oboe/samples/RhythmGame/third_party/glm/gtc/ |
| D | noise.inl | 16 T pW = static_cast<T>(1.5) - dot(abs(pXYZ), tvec3<T, P>(1)); local
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/ |
| D | QuantizedOpKernels.cpp | 1464 int64_t pW, // padding in _qmaxpool_2d_nhwc_kernel()
1564 int64_t pW, // padding in qmaxpool_2d_nhwc_kernel()
1596 int64_t pW, // padding in qmaxpool_3d_nthwc_kernel()
|
| /external/pytorch/torch/ |
| D | _meta_registrations.py | 3944 pW, argument
|