Searched refs:small_dimension_size (Results 1 – 1 of 1) sorted by relevance
856 int64_t small_dimension_size = smaller_shape.dimensions(i); in InferInDimBroadcastShape() local864 if (small_dimension_size != large_dimension_size && in InferInDimBroadcastShape()865 small_dimension_size != 1 && large_dimension_size != 1) { in InferInDimBroadcastShape()868 small_dimension_size, large_dimension_size, in InferInDimBroadcastShape()873 if (small_dimension_size == large_dimension_size || in InferInDimBroadcastShape()874 (small_dimension_size == 1 && !small_is_dynamic) || in InferInDimBroadcastShape()892 output_shape.set_dimensions(dimension_to_match, small_dimension_size); in InferInDimBroadcastShape()