Searched refs:padded_concat_shape (Results 1 – 1 of 1) sorted by relevance
585 auto padded_concat_shape = ShapeInference::InferPadShape( in PadFromPartialReplicateShape() local589 padded_concat_shape, concat, zero, pad_config)); in PadFromPartialReplicateShape()967 auto padded_concat_shape = concat->shape(); in ExchangeHaloAndGetValidData() local978 padded_concat_shape.set_dimensions(dim, concat->shape().dimensions(dim) + in ExchangeHaloAndGetValidData()983 padded_concat_shape, concat, pad_value, padding_config)); in ExchangeHaloAndGetValidData()