Searched refs:manual_to_onedevice (Results 1 – 1 of 1) sorted by relevance
1412 auto manual_to_onedevice = [&](const Shape& shape, in Preprocess() local1439 manual_to_onedevice(hlo->shape(), *visiting_hlo_sharding_)); in Preprocess()1445 manual_to_onedevice(operand->shape(), operand->sharding())); in Preprocess()