@jim19930609: This interface will be further modified until we finalized the design for Allocator - Allocation For now, we have to temporarily a

In this function, dict is not viewed as sequence.

Return all variable nodes included in the graph as a set.

Convert the data type in numpy to the data type in Paddle. Args: np_dtype (np.dtype|str): The data type in numpy or valid data type

Create a operator node in the graph. Args: op_type(str): the type of the operator node. attrs(dict): the att

Create a persistable variable node in the graph. In IrGraph, it can not distinguish between persistable variables and parameters.

count FLOPs for operation. Args: op_type (str): the type of operation. input_shapes (dict): the shapes of inputs. at

r""" It computes the pairwise distance between two vectors. The distance is calculated by p-order norm: .. math:: \Vert x \Vert

Checks if all work currently captured by event has completed. Returns: bool: Whether all work currently captured by eve

Change if autograd should record operations on parameters in this layer. Parameters: requires_grad (bool): whether autog

A wrapper for Optimizer's step function to finish the update operation of the optimizer.

TODO(Ruting): modify name to GetNullableSingleInputGrad after Large-scale development

NOTE(zhangbo): Some operators of Paddle support optional inputs or outputs, representing whether the input or output exists. In the Pir, whether the v