xref: /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/Factory.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1 #include <ATen/native/vulkan/ops/Common.h>
2 
3 namespace at {
4 namespace native {
5 namespace vulkan {
6 namespace ops {
7 
8 Tensor _empty_affine_quantized(
9     const IntArrayRef sizes,
10     const std::optional<ScalarType> dtype,
11     const std::optional<c10::Layout> layout,
12     const std::optional<Device> device,
13     const std::optional<bool> pin_memory,
14     const double scale,
15     const int64_t zero_point,
16     const std::optional<MemoryFormat> memory_format);
17 
18 } // namespace ops
19 } // namespace vulkan
20 } // namespace native
21 } // namespace at
22