Searched defs:ShapesAreBroadcastCompatible (Results 1 – 1 of 1) sorted by relevance
154 struct ShapesAreBroadcastCompatible : public Rule struct156 unsigned int CalcInputSize(const TensorShape& in, const TensorShape& out, unsigned int idx) in CalcInputSize()163 … ShapesAreBroadcastCompatible(const TensorInfo& in0, const TensorInfo& in1, const TensorInfo& out) in ShapesAreBroadcastCompatible() argument