Searched refs:ChunkAlias (Results 1 – 4 of 4) sorted by relevance
46 virtual Tensor ChunkAlias(int i) = 0;
105 Tensor alias_chunk(ca_->ChunkAlias(col_params_->subdiv_rank[0])); in Run()
142 Tensor ChunkAlias(int i) override { in ChunkAlias() function in tensorflow::__anon2cb66ace0211::CollectiveAdapterImpl
346 rf->chunk = ca_->ChunkAlias(rf->sc_idx); in InitRingField()