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
306 const int block_rank = total_dims - original_rank; in VerifyAndCountSparseElements() local
66 int block_rank = 0; in PopulateEncodingParams() local
839 int64_t block_rank = block_shape_type.getNumElements(); in matchAndRewrite() local1055 const int block_rank = block_shape_ty.getShape().front(); in matchAndRewrite() local
826 int block_rank = block_shape[0]; in convertSpaceToBatchNDOp() local1077 int block_rank = block_shape_type.getShape()[0]; in convertBatchToSpaceNDOp() local
1226 const int64_t block_rank = in Verify() local1318 const int64_t block_rank = in inferReturnTypes() local
428 const int block_rank = block_shape_ty.getShape().front(); in Verify() local