1 #pragma once 2 3 #include <torch/csrc/lazy/core/tensor.h> 4 5 namespace torch { 6 namespace lazy { 7 8 ////////////////////////////////////////////////////////////////////////////// 9 // ATEN operators follows here, listed in alphabetical order. 10 ////////////////////////////////////////////////////////////////////////////// 11 12 void copy_(torch::lazy::LazyTensorPtr& input, torch::lazy::LazyTensorPtr& src); 13 // Fills the input with the given value. 14 void fill_(torch::lazy::LazyTensorPtr& input, const at::Scalar& value); 15 16 } // namespace lazy 17 } // namespace torch 18