Home
last modified time | relevance | path

Searched refs:NewChannel (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dchannel_tracker.h57 StatusOr<ChannelHandle> NewChannel(ChannelHandle::ChannelType type);
Dchannel_tracker.cc34 StatusOr<ChannelHandle> ChannelTracker::NewChannel( in NewChannel() function in xla::ChannelTracker
Dservice.cc189 channel_tracker_.NewChannel(arg->channel_type())); in CreateChannelHandle()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h326 int64 NewChannel() const { return (*state_.next_channel_id)++; } in NewChannel() function
480 int64 NewChannel() { return (*next_channel_id_)++; } in NewChannel() function
Dspmd_partitioner.cc1081 state_.b, operand, reduction, {}, NewChannel()); in Broadcast()
1565 {}, NewChannel()); in HandleConcatenate()