Searched refs:expand_dims_without_halo_exchange (Results 1 – 1 of 1) sorted by relevance
522 std::vector<int64> expand_dims_without_halo_exchange; in PadFromPartialReplicateShape() local541 expand_dims_without_halo_exchange.emplace_back(dim); in PadFromPartialReplicateShape()609 if (!expand_dims_without_halo_exchange.empty()) { in PadFromPartialReplicateShape()614 for (auto dim : expand_dims_without_halo_exchange) { in PadFromPartialReplicateShape()