Merge two symbolDim if they are compatible.

Helper to construct a NodeDef.

static */

Returns the number of elements in a cluster.

Skip the specified number of samples of 128-bits in the current stream.

Returns true if a tensor is constant and all its elements are equal to `elemVal`

Gives a consistent order of a list op SymbolicDim Ops

Create a Canonicalizer pass.

reinterpret_cast the input memref into 1D

Get full path of dumped file".

Fill the tensor with a src tensor TODO(xpz): may replace is_deconv_weights with a enum for other purposes

Convert to a fusion type from its string representation.

Returns the full name of the fusion op Here full name is composed of the name and tag of the fusion op.

Returns the fusion kind of the fusion pattern.

Adds related depedent dialects (e.g. PDL dialect).

returns the users of the `memref`. The users should be in the same fusion like `op`.

Get indexes of input nodes for node of given index.

Find tensors that come into graph segment as input.

Returns true if the underlying buffer of this memref is a const buffer.

Check Op if it is a mhlo Op or tensor Op.

Check Op if it is a mhlo Op.

Returns true if the fusion type is transform-based fusion.

Iterate over the submodules and return name list of those whose type is matched in type_list. Args: model (nn.Module): Model to be m

return a new memref type with provided memory space if the input type if a memref type otherwise return the original type.