Lines Matching refs:pt
11 from tools import mlir_pytaco_api as pt unknown
17 compressed = pt.compressed
18 dense = pt.dense
21 csr = pt.format([dense, compressed])
22 dv = pt.format([dense])
30 A = pt.read(os.path.join(_SCRIPT_PATH, "data/pwtk.mtx"), csr)
34 x = pt.from_array(np.full((A.shape[1],), 1, dtype=np.float32))
35 z = pt.from_array(np.full((A.shape[0],), 2, dtype=np.float32))
38 y = pt.tensor([A.shape[0]], dv)
41 i, j = pt.get_index_vars(2)
52 pt.write(out_file, y)