Etusivu Tutki Apua
Kirjaudu sisään
SRI-DYZBC2
/
Vehicle-cpp
2
0
Fork 0
Tiedostot Ongelmat 0 Pull-pyynnöt 0 Wiki
Puu: e6b2cb2e99
Haarat Tagit
Vehicle-cpp
/
hkpc
/
software
/
torch
/
include
/
ATen
/
ops
/
scatter_meta_dispatch.h

scatter_meta_dispatch.h 2.9 KB
Historia Raaka

1234567891011121314151617181920212223242526272829303132333435363738 
  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 meta {
    19. 

  19. 

  20. TORCH_API at::Tensor scatter(const at::Tensor & self, int64_t dim, const at::Tensor & index, const at::Tensor & src);
    21. 

  21. TORCH_API at::Tensor & scatter_out(at::Tensor & out, const at::Tensor & self, int64_t dim, const at::Tensor & index, const at::Tensor & src);
    22. 

  22. TORCH_API at::Tensor & scatter_outf(const at::Tensor & self, int64_t dim, const at::Tensor & index, const at::Tensor & src, at::Tensor & out);
    23. 

  23. TORCH_API at::Tensor & scatter_(at::Tensor & self, int64_t dim, const at::Tensor & index, const at::Tensor & src);
    24. 

  24. TORCH_API at::Tensor scatter(const at::Tensor & self, int64_t dim, const at::Tensor & index, const at::Scalar & value);
    25. 

  25. TORCH_API at::Tensor & scatter_out(at::Tensor & out, const at::Tensor & self, int64_t dim, const at::Tensor & index, const at::Scalar & value);
    26. 

  26. TORCH_API at::Tensor & scatter_outf(const at::Tensor & self, int64_t dim, const at::Tensor & index, const at::Scalar & value, at::Tensor & out);
    27. 

  27. TORCH_API at::Tensor & scatter_(at::Tensor & self, int64_t dim, const at::Tensor & index, const at::Scalar & value);
    28. 

  28. TORCH_API at::Tensor scatter(const at::Tensor & self, int64_t dim, const at::Tensor & index, const at::Tensor & src, c10::string_view reduce);
    29. 

  29. TORCH_API at::Tensor & scatter_out(at::Tensor & out, const at::Tensor & self, int64_t dim, const at::Tensor & index, const at::Tensor & src, c10::string_view reduce);
    30. 

  30. TORCH_API at::Tensor & scatter_outf(const at::Tensor & self, int64_t dim, const at::Tensor & index, const at::Tensor & src, c10::string_view reduce, at::Tensor & out);
    31. 

  31. TORCH_API at::Tensor & scatter_(at::Tensor & self, int64_t dim, const at::Tensor & index, const at::Tensor & src, c10::string_view reduce);
    32. 

  32. TORCH_API at::Tensor scatter(const at::Tensor & self, int64_t dim, const at::Tensor & index, const at::Scalar & value, c10::string_view reduce);
    33. 

  33. TORCH_API at::Tensor & scatter_out(at::Tensor & out, const at::Tensor & self, int64_t dim, const at::Tensor & index, const at::Scalar & value, c10::string_view reduce);
    34. 

  34. TORCH_API at::Tensor & scatter_outf(const at::Tensor & self, int64_t dim, const at::Tensor & index, const at::Scalar & value, c10::string_view reduce, at::Tensor & out);
    35. 

  35. TORCH_API at::Tensor & scatter_(at::Tensor & self, int64_t dim, const at::Tensor & index, const at::Scalar & value, c10::string_view reduce);
    36. 

  36. 

  37. } // namespace meta
    38. 

  38. } // namespace at
    39.