Searched refs:sharded_pad (Results 1 – 1 of 1) sorted by relevance
2663 auto sharded_pad = reshard_operand->sharded_input; in HandlePad() local2666 ShapeInference::InferPadShape(sharded_pad->shape(), in HandlePad()2669 sharded_pad = b_.AddInstruction(hlo->CreatePad(sharded_pad_shape, in HandlePad()2670 sharded_pad, replicated_rhs, in HandlePad()2676 return sharded_pad; in HandlePad()2680 shard_shape, sharded_pad, in HandlePad()