Home
last modified time | relevance | path

Searched refs:GetPromotedBinaryOpShape (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/ops/
H A Darithmetic_ir_ops.cpp21 GetPromotedBinaryOpShape(node1.shape(), node2.shape())); in operator +()
28 GetPromotedBinaryOpShape(node1.shape(), node2.shape())); in operator -()
35 GetPromotedBinaryOpShape(node1.shape(), node2.shape())); in operator *()
42 GetPromotedBinaryOpShape(node1.shape(), node2.shape())); in operator /()
/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A Dhelpers.h67 GetPromotedBinaryOpShape(const Shape& shape1, const Shape& shape2);
H A Dhelpers.cpp123 Shape GetPromotedBinaryOpShape(const Shape& shape1, const Shape& shape2) { in GetPromotedBinaryOpShape() function