Searched refs:maybe_input_sharded_dim (Results 1 – 1 of 1) sorted by relevance
2010 auto maybe_input_sharded_dim = UniqueTiledDim(operand.sharding()); in HandleReshape() local2012 if (!maybe_input_sharded_dim || !maybe_output_sharded_dim) { in HandleReshape()2015 int64 input_sharded_dim = *maybe_input_sharded_dim; in HandleReshape()