/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/tensorexpr/operators
NameSizeModeActions
conv2d.h11820644editdlrm
matmul.h5070644editdlrm
norm.h3500644editdlrm
operators.h3510644editdlrm
reduction.h6570644editdlrm
softmax.h3210644editdlrm
unary.h3060644editdlrm
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/tensorexpr/operators/unary.h (306B)
#pragma once #include namespace torch { namespace jit { namespace tensorexpr { TORCH_API Tensor computeSign( const std::vector& inputs, const std::vector& outputShape); } // namespace tensorexpr } // namespace jit } // namespace torch