Searched defs:IsBatchParallel (Results 1 – 1 of 1) sorted by relevance
920 bool Layout::IsBatchParallel() const { in IsBatchParallel() function in tensorflow::dtensor::Layout935 bool Layout::IsBatchParallel(int non_batch_rank) const { in IsBatchParallel() function in tensorflow::dtensor::Layout