Searched refs:larger_shape (Results 1 – 2 of 2) sorted by relevance
744 const Shape& smaller_shape, const Shape& larger_shape, in InferInDimBroadcastShape() argument752 ShapeUtil::HumanString(larger_shape)); in InferInDimBroadcastShape()800 Shape output_shape(larger_shape); in InferInDimBroadcastShape()802 ShapeUtil::HigherPrecisionElementType(larger_shape, smaller_shape)); in InferInDimBroadcastShape()811 if (dimension_to_match >= larger_shape.dimensions_size()) { in InferInDimBroadcastShape()815 dimension_to_match, larger_shape.dimensions_size()); in InferInDimBroadcastShape()818 int64 large_dimension_size = larger_shape.dimensions(dimension_to_match); in InferInDimBroadcastShape()821 larger_shape.is_dynamic_dimension(dimension_to_match); in InferInDimBroadcastShape()831 ShapeUtil::HumanString(larger_shape)); in InferInDimBroadcastShape()842 ShapeUtil::HumanString(larger_shape)); in InferInDimBroadcastShape()[all …]
333 const Shape& smaller_shape, const Shape& larger_shape,