Abstract: Apparatuses, systems, and techniques to perform one or more operations using a tensor. In at least one embodiment, one or more circuits are to perform an application programming interface (API) to cause parameters indicating operands and dimensions of operands of two or more tensor contractions to be stored.

Abstract: Apparatuses, systems, and techniques to perform one or more operations using a tensor. In at least one embodiment, one or more circuits are to perform an application programming interface (API) to cause two or more tensor contractions to be performed based, at least in part, on one or more input parameters of the API.

Abstract: In various examples, systems and methods are disclosed relating to generating segmented tensor products using graphics processing units. One or more circuits can identify a plurality of operands for a tensor operation. Each operand divided into a set of segments each including at least a portion of the operand. The one or more circuits can identify a plurality of paths for the tensor operation, each path identifying a set of coefficients and at least one segment of each operand of the plurality of operands. The one or more circuits can generate a respective partial output for each path of the plurality of paths according to the tensor operation, the at least one segment of the path, and the set of coefficients of the path. The one or more circuits can generate an output tensor for the tensor operation based at least on the respective partial output for each of the plurality of paths.

Abstract: Apparatuses, systems, and techniques to perform one or more operations using a tensor. In at least one embodiment, one or more circuits are to perform an application programming interface (API) to perform one or more operations using one or more tensors based on at least one or more indications of the plurality of operations by the API, one or more compiler options indicated by the API, and/or one of one or more indications of an amount of storage to be used to perform the one or more operations.

Abstract: Apparatuses, systems, and techniques to perform one or more operations using a tensor. In at least one embodiment, one or more circuits are to perform an application programming interface (API) to perform one or more operations using one or more tensors based on at least one or more indications of the plurality of operations by the API, one or more compiler options indicated by the API, and/or one of one or more indications of an amount of storage to be used to perform the one or more operations.

Abstract: Apparatuses, systems, and techniques to perform one or more operations using a tensor. In at least one embodiment, one or more circuits are to perform an application programming interface (API) to perform one or more operations using one or more tensors based on at least one or more indications of the plurality of operations by the API, one or more compiler options indicated by the API, and/or one of one or more indications of an amount of storage to be used to perform the one or more operations.

Abstract: Apparatuses, systems, and techniques to store tensor operands. In at least one embodiment, modes of one or more tensor operands are sorted based, at least in part, on one or more performance metrics of one or more tensor operations to be performed using said one or more tensor operands.