Home
last modified time | relevance | path

Searched refs:custom_add_Tensor (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/test/inductor/extension_backends/cpp/
H A Dextension_device.cpp27 at::Tensor custom_add_Tensor(const at::Tensor & self, const at::Tensor & other, const at::Scalar & … in custom_add_Tensor() function
155 m.impl("add.Tensor", &custom_add_Tensor); in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/pytorch/test/cpp_extensions/
H A Dopen_registration_extension.cpp246 at::Tensor custom_add_Tensor(const at::Tensor& self, const at::Tensor& other, const at::Scalar& alp… in custom_add_Tensor() function
508 m.impl("add.Tensor", &custom_add_Tensor); in TORCH_LIBRARY_IMPL()