Searched refs:ConstrainChannel (Results 1 – 2 of 2) sorted by relevance
269 const Layout* ConstrainChannel(int64 channel_id, const Layout& layout) { in ConstrainChannel() function
407 const Layout* prev_layout = host_channel_constraints_.ConstrainChannel( in BuildHostChannelConstraints()1822 ->ConstrainChannel( in ConstrainChannelLayouts()1838 ->ConstrainChannel(instruction->channel_id(), in ConstrainChannelLayouts()1863 ->ConstrainChannel(instruction->all_reduce_id().value(), in ConstrainChannelLayouts()