xref: /aosp_15_r20/external/pytorch/torch/csrc/lazy/core/ops/arithmetic_ir_ops.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1 #pragma once
2 
3 #include <torch/csrc/lazy/core/ir.h>
4 
5 namespace torch {
6 namespace lazy {
7 
8 TORCH_API NodePtr operator+(const Value& node1, const Value& node2);
9 TORCH_API NodePtr operator-(const Value& node1, const Value& node2);
10 TORCH_API NodePtr operator*(const Value& node1, const Value& node2);
11 TORCH_API NodePtr operator/(const Value& node1, const Value& node2);
12 
13 } // namespace lazy
14 } // namespace torch
15