Searched refs:replicate_on_last_tile_dim_ (Results 1 – 2 of 2) sorted by relevance
156 bool ReplicateOnLastTileDim() const { return replicate_on_last_tile_dim_; } in ReplicateOnLastTileDim()248 replicate_on_last_tile_dim_ == other.replicate_on_last_tile_dim_;299 replicate_on_last_tile_dim_(false),313 replicate_on_last_tile_dim_(false),323 replicate_on_last_tile_dim_(replicate_on_last_tile_dim), in replicated_()332 replicate_on_last_tile_dim_(false) {}373 bool replicate_on_last_tile_dim_; variable
193 replicate_on_last_tile_dim_ ? " last_tile_dim_replicate" : "", in ToString()241 if (replicate_on_last_tile_dim_) { in TileIndexForDevice()254 if (replicate_on_last_tile_dim_ && in DeviceForTileIndex()271 if (replicate_on_last_tile_dim_) { in TileOffsetForDevice()455 if (shape.rank() + (replicate_on_last_tile_dim_ ? 1 : 0) != in ValidateNonTuple()708 if (replicate_on_last_tile_dim_) { in Hash()