Searched refs:batchify (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/aten/src/ATen/native/ |
| D | Convolution.cpp | 756 static std::tuple<Tensor, bool> batchify( in batchify() function
930 auto [input, is_batched] = batchify(input_, /*num_spatial_dims=*/ 1, "conv1d"); in conv1d_symint()
955 auto [input, is_batched] = batchify(input_, /*num_spatial_dims=*/ 2, "conv2d"); in conv2d_symint()
980 auto [input, is_batched] = batchify(input_, /*num_spatial_dims=*/ 3, "conv3d"); in conv3d_symint()
1078 auto [input, is_batched] = batchify(input_, /*num_spatial_dims=*/ 1, "conv1d"); in conv1d_padding_symint()
1092 auto [input, is_batched] = batchify(input_, /*num_spatial_dims=*/ 2, "conv2d"); in conv2d_padding_symint()
1106 auto [input, is_batched] = batchify(input_, /*num_spatial_dims=*/ 3, "conv3d"); in conv3d_padding_symint()
1123 auto [input, is_batched] = batchify(input_, /*num_spatial_dims=*/ 1, "conv_transpose1d"); in conv_transpose1d_symint()
1142 auto [input, is_batched] = batchify(input_, /*num_spatial_dims=*/ 2, "conv_transpose2d"); in conv_transpose2d_symint()
1161 auto [input, is_batched] = batchify(input_, /*num_spatial_dims=*/ 3, "conv_transpose3d"); in conv_transpose3d_symint()
|