Strona główna Odkrywaj Pomoc
Zaloguj się
SRI-DYZBC2
/
Vehicle-cpp
2
0
Forkuj 0
Pliki Problemy 0 Oczekujące zmiany 0 Wiki
Drzewo: e6b2cb2e99
Gałęzie Tagi
Vehicle-cpp
/
hkpc
/
software
/
torch
/
include
/
ATen
/
ops
/
rand_compositeexplicitautograd_dispatch.h

rand_compositeexplicitautograd_dispatch.h 4.8 KB
Historia Czysty

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950 
  1. #pragma once
    2. 

  2. // @generated by torchgen/gen.py from DispatchKeyFunction.h
    3. 

  3. 

  4. // NB: The implementing C++ file is RegisterDispatchKey.cpp
    5. 

  5. 

  6. // The only #includes we need are for custom classes that have defaults in the C++ API
    7. 

  7. #include <c10/core/MemoryFormat.h>
    8. 

  8. #include <c10/core/Scalar.h>
    9. 

  9. #include <ATen/core/Reduction.h>
    10. 

  10. 

  11. // Forward declarations of any types needed in the operator signatures.
    12. 

  12. // We can't directly include these classes because it will cause circular include dependencies.
    13. 

  13. // This file is included by TensorBody.h, which defines the Tensor class.
    14. 

  14. #include <ATen/core/ATen_fwd.h>
    15. 

  15. 

  16. namespace at {
    17. 

  17. 

  18. namespace compositeexplicitautograd {
    19. 

  19. 

  20. TORCH_API at::Tensor rand(at::IntArrayRef size, c10::optional<at::DimnameList> names, at::TensorOptions options={});
    21. 

  21. TORCH_API at::Tensor rand(at::IntArrayRef size, c10::optional<at::DimnameList> names, c10::optional<at::ScalarType> dtype, c10::optional<at::Layout> layout, c10::optional<at::Device> device, c10::optional<bool> pin_memory);
    22. 

  22. TORCH_API at::Tensor rand_symint(c10::SymIntArrayRef size, c10::optional<at::DimnameList> names, at::TensorOptions options={});
    23. 

  23. TORCH_API at::Tensor rand_symint(c10::SymIntArrayRef size, c10::optional<at::DimnameList> names, c10::optional<at::ScalarType> dtype, c10::optional<at::Layout> layout, c10::optional<at::Device> device, c10::optional<bool> pin_memory);
    24. 

  24. TORCH_API at::Tensor & rand_out(at::Tensor & out, at::IntArrayRef size, c10::optional<at::DimnameList> names);
    25. 

  25. TORCH_API at::Tensor & rand_outf(at::IntArrayRef size, c10::optional<at::DimnameList> names, at::Tensor & out);
    26. 

  26. TORCH_API at::Tensor & rand_symint_out(at::Tensor & out, c10::SymIntArrayRef size, c10::optional<at::DimnameList> names);
    27. 

  27. TORCH_API at::Tensor & rand_symint_outf(c10::SymIntArrayRef size, c10::optional<at::DimnameList> names, at::Tensor & out);
    28. 

  28. TORCH_API at::Tensor rand(at::IntArrayRef size, c10::optional<at::Generator> generator, c10::optional<at::DimnameList> names, at::TensorOptions options={});
    29. 

  29. TORCH_API at::Tensor rand(at::IntArrayRef size, c10::optional<at::Generator> generator, c10::optional<at::DimnameList> names, c10::optional<at::ScalarType> dtype, c10::optional<at::Layout> layout, c10::optional<at::Device> device, c10::optional<bool> pin_memory);
    30. 

  30. TORCH_API at::Tensor rand_symint(c10::SymIntArrayRef size, c10::optional<at::Generator> generator, c10::optional<at::DimnameList> names, at::TensorOptions options={});
    31. 

  31. TORCH_API at::Tensor rand_symint(c10::SymIntArrayRef size, c10::optional<at::Generator> generator, c10::optional<at::DimnameList> names, c10::optional<at::ScalarType> dtype, c10::optional<at::Layout> layout, c10::optional<at::Device> device, c10::optional<bool> pin_memory);
    32. 

  32. TORCH_API at::Tensor & rand_out(at::Tensor & out, at::IntArrayRef size, c10::optional<at::Generator> generator, c10::optional<at::DimnameList> names);
    33. 

  33. TORCH_API at::Tensor & rand_outf(at::IntArrayRef size, c10::optional<at::Generator> generator, c10::optional<at::DimnameList> names, at::Tensor & out);
    34. 

  34. TORCH_API at::Tensor & rand_symint_out(at::Tensor & out, c10::SymIntArrayRef size, c10::optional<at::Generator> generator, c10::optional<at::DimnameList> names);
    35. 

  35. TORCH_API at::Tensor & rand_symint_outf(c10::SymIntArrayRef size, c10::optional<at::Generator> generator, c10::optional<at::DimnameList> names, at::Tensor & out);
    36. 

  36. TORCH_API at::Tensor rand(at::IntArrayRef size, at::TensorOptions options={});
    37. 

  37. TORCH_API at::Tensor rand(at::IntArrayRef size, c10::optional<at::ScalarType> dtype, c10::optional<at::Layout> layout, c10::optional<at::Device> device, c10::optional<bool> pin_memory);
    38. 

  38. TORCH_API at::Tensor rand_symint(c10::SymIntArrayRef size, at::TensorOptions options={});
    39. 

  39. TORCH_API at::Tensor rand_symint(c10::SymIntArrayRef size, c10::optional<at::ScalarType> dtype, c10::optional<at::Layout> layout, c10::optional<at::Device> device, c10::optional<bool> pin_memory);
    40. 

  40. TORCH_API at::Tensor & rand_out(at::Tensor & out, at::IntArrayRef size);
    41. 

  41. TORCH_API at::Tensor & rand_outf(at::IntArrayRef size, at::Tensor & out);
    42. 

  42. TORCH_API at::Tensor & rand_symint_out(at::Tensor & out, c10::SymIntArrayRef size);
    43. 

  43. TORCH_API at::Tensor & rand_symint_outf(c10::SymIntArrayRef size, at::Tensor & out);
    44. 

  44. TORCH_API at::Tensor rand(at::IntArrayRef size, c10::optional<at::Generator> generator, at::TensorOptions options={});
    45. 

  45. TORCH_API at::Tensor rand(at::IntArrayRef size, c10::optional<at::Generator> generator, c10::optional<at::ScalarType> dtype, c10::optional<at::Layout> layout, c10::optional<at::Device> device, c10::optional<bool> pin_memory);
    46. 

  46. TORCH_API at::Tensor rand_symint(c10::SymIntArrayRef size, c10::optional<at::Generator> generator, at::TensorOptions options={});
    47. 

  47. TORCH_API at::Tensor rand_symint(c10::SymIntArrayRef size, c10::optional<at::Generator> generator, c10::optional<at::ScalarType> dtype, c10::optional<at::Layout> layout, c10::optional<at::Device> device, c10::optional<bool> pin_memory);
    48. 

  48. 

  49. } // namespace compositeexplicitautograd
    50. 

  50. } // namespace at
    51.