Searched refs:dimension_to_match (Results 1 – 1 of 1) sorted by relevance
840 int64 dimension_to_match = broadcast_dimensions.at(i); in InferInDimBroadcastShape() local841 if (dimension_to_match < 0) { in InferInDimBroadcastShape()844 dimension_to_match); in InferInDimBroadcastShape()846 if (dimension_to_match >= larger_shape.dimensions_size()) { in InferInDimBroadcastShape()850 dimension_to_match, larger_shape.dimensions_size()); in InferInDimBroadcastShape()853 int64 large_dimension_size = larger_shape.dimensions(dimension_to_match); in InferInDimBroadcastShape()856 larger_shape.is_dynamic_dimension(dimension_to_match); in InferInDimBroadcastShape()882 if (i > 0 && broadcast_dimensions.at(i - 1) >= dimension_to_match) { in InferInDimBroadcastShape()885 dimension_to_match, broadcast_dimensions.at(i - 1)); in InferInDimBroadcastShape()888 output_shape.set_dimensions(dimension_to_match, small_dimension_size); in InferInDimBroadcastShape()[all …]