Searched refs:large_dimension_size (Results 1 – 1 of 1) sorted by relevance
818 int64 large_dimension_size = larger_shape.dimensions(dimension_to_match); in InferInDimBroadcastShape() local825 if (small_dimension_size != large_dimension_size && in InferInDimBroadcastShape()826 small_dimension_size != 1 && large_dimension_size != 1) { in InferInDimBroadcastShape()829 small_dimension_size, large_dimension_size, in InferInDimBroadcastShape()834 if (small_dimension_size == large_dimension_size || in InferInDimBroadcastShape()836 (large_dimension_size == 1 && !large_is_dynamic)) { in InferInDimBroadcastShape()