Searched refs:should_decompose (Results 1 – 1 of 1) sorted by relevance
30 std::function<bool(const HloAllGatherInstruction&)> should_decompose) in AllGatherDecomposer() argument31 : should_decompose_(std::move(should_decompose)) {} in AllGatherDecomposer()