Searched defs:new_batch_size (Results 1 – 1 of 1) sorted by relevance
750 const int64_t new_batch_size = in SplitAndTransposeMergedBatch() local2296 const int64_t new_batch_size = new_shape.dimensions(new_batch_dim); in SelectValidPortion() local2980 const int64_t new_batch_size = old_batch_size * ctrl_.number_of_splits; in PropagateOnConstant() local