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