Searched refs:batch_cst (Results 1 – 1 of 1) sorted by relevance
1345 int batch_cst = input_ty.getShape()[0]; in matchAndRewrite() local1487 RankedTensorType::get({batch_cst, in_spatial_cst, channels_cst}, in matchAndRewrite()1496 RankedTensorType::get({batch_cst, out_spatial_cst, channels_cst}, in matchAndRewrite()