Searched refs:return_batch (Results 1 – 1 of 1) sorted by relevance
1340 int64_t return_batch = input_shape[0]; in inferReturnTypes() local1344 return_batch = ShapedType::kDynamicSize; in inferReturnTypes()1346 if (return_batch == ShapedType::kDynamicSize) { in inferReturnTypes()1355 return_batch *= block_shape_i; in inferReturnTypes()1358 return_shape[0] = return_batch; in inferReturnTypes()