Searched refs:channels_cst (Results 1 – 1 of 1) sorted by relevance
1346 int channels_cst = input_ty.getShape()[3]; 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()