Searched refs:dynamic_dim_index (Results 1 – 2 of 2) sorted by relevance
1595 llvm::Optional<int> dynamic_dim_index; in verify() local1606 if (dynamic_dim_index) in verify()1609 dynamic_dim_index = dim.index(); in verify()1615 if (!dynamic_dim_index && total_dim_size != input_dim_size) in verify()1621 if (dynamic_dim_index && total_dim_size > input_dim_size) in verify()
3513 llvm::Optional<int> dynamic_dim_index; in matchAndRewrite() local3521 assert(!dynamic_dim_index && "invalid split sizes"); in matchAndRewrite()3522 dynamic_dim_index = dim.index(); in matchAndRewrite()3536 assert(((dynamic_dim_index && total_dim_size <= input_dim_size) || in matchAndRewrite()3537 (!dynamic_dim_index && total_dim_size == input_dim_size)) && in matchAndRewrite()3541 if (dynamic_dim_index) in matchAndRewrite()3542 split_sizes[*dynamic_dim_index] = input_dim_size - total_dim_size; in matchAndRewrite()