Searched defs:block_rank (Results 1 – 8 of 8) sorted by relevance
31 const int block_rank = block_shape.size(); in BatchToSpace() local
31 const int block_rank = block_shape.size(); in SpaceToBatch() local
298 const int block_rank = total_dims - original_rank; in VerifyAndCountSparseElements() local
66 int block_rank = 0; in PopulateEncodingParams() local
880 int64_t block_rank = block_shape_type.getNumElements(); in matchAndRewrite() local1093 const int block_rank = block_shape_ty.getShape().front(); in matchAndRewrite() local
734 int block_rank = block_shape[0]; in convertSpaceToBatchNDOp() local984 int block_rank = block_shape_type.getShape()[0]; in convertBatchToSpaceNDOp() local
1352 const int64_t block_rank = in Verify() local
428 const int block_rank = block_shape_ty.getShape().front(); in Verify() local