Searched refs:IsBatchParallel (Results 1 – 4 of 4) sorted by relevance
58 if (input_layout.IsBatchParallel()) in VerifyConvLayout()150 if (input_layout.IsBatchParallel()) in HandleConv()320 if (input_layout.IsBatchParallel()) { in HandleConvBackpropFilter()401 if (!output_layout->IsFullyReplicated() && !output_layout->IsBatchParallel()) in ExpandOp()
61 if (!layouts[i].IsBatchParallel(batchable_indices.lookup(i))) return false; in AllBatchParallel()
302 bool IsBatchParallel() const;304 bool IsBatchParallel(int non_batch_rank) const;
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