Searched defs:full_batch (Results 1 – 13 of 13) sorted by relevance
52 bool full_batch = ParallelContext::GetInstance()->full_batch(); in GenerateOpStrategies() local
120 bool full_batch = ParallelContext::GetInstance()->full_batch(); in CheckStrategy() local
271 bool full_batch = ParallelContext::GetInstance()->full_batch(); in GenerateOpStrategies() local
34 resize_width=224, full_batch=False, scale=1.0, rank_size=1): argument
105 bool full_batch = parallel::ParallelContext::GetInstance()->full_batch(); in GetFirstEmbeddingCacheTableInfo() local
275 bool full_batch = parallel::ParallelContext::GetInstance()->full_batch(); in ParallelSymbolInfo() local
140 void ParallelContext::set_full_batch(bool full_batch) { in set_full_batch()
79 bool full_batch() const { return full_batch_; } in full_batch() function
495 def set_full_batch(self, full_batch): argument
313 bool full_batch = ParallelContext::GetInstance()->full_batch(); in SetStridedSliceStrategy() local
397 bool full_batch = ParallelContext::GetInstance()->full_batch(); in HandleNoUsedParameter() local
1875 bool full_batch = ParallelContext::GetInstance()->full_batch(); in SetVirtualDatasetStrategy() local
1326 bool full_batch = ParallelContext::GetInstance()->full_batch(); in CreateZeroseOutput() local