Searched refs:tmp_reshape (Results 1 – 1 of 1) sorted by relevance
2096 auto tmp_reshape = b_.AddInstruction( in HandleReshape() local2098 tmp_reshape->set_metadata(hlo->metadata()); in HandleReshape()2099 tmp_reshape->set_sharding(hlo->sharding()); in HandleReshape()2105 PartitionedHlo(tmp_reshape, tmp_full_shape, MakePartitioningState()); in HandleReshape()