Searched refs:expected_batch_size (Results 1 – 4 of 4) sorted by relevance
825 ClusterBatchSize expected_batch_size = in SegmentGraph() local829 VLOG(3) << "batch size " << expected_batch_size; in SegmentGraph()850 ClusterBatchSize merged_batch_size = expected_batch_size; in SegmentGraph()853 << expected_batch_size.ToString() << " " in SegmentGraph()872 expected_batch_size = merged_batch_size; in SegmentGraph()905 if (expected_batch_size != in SegmentGraph()
203 expected_batch_size = 5 if is_even else 10204 self.assertEqual(expected_batch_size, result.shape[0])
3239 expected_batch_size = None3242 if expected_batch_size is None:3244 expected_batch_size = tensors[i].shape.dims[0]3245 elif not expected_batch_size.is_compatible_with(tensors[i].shape.dims[0]):3250 expected_batch_size, tensors[i].shape.dims[0]))
4299 expected_batch_size = None4304 if expected_batch_size is None:4306 expected_batch_size = batch_size4307 elif not expected_batch_size.is_compatible_with(batch_size):4312 expected_batch_size, batch_size))