Searched defs:smaller_shape (Results 1 – 2 of 2) sorted by relevance
119 def _itGen(self, smaller_shape, larger_shape): argument
779 const Shape& smaller_shape, const Shape& larger_shape, in InferInDimBroadcastShape()939 const Shape& smaller_shape = lhs.rank() > rhs.rank() ? rhs : lhs; in InferElementwiseBinaryOpShape() local