Lines Matching refs:pt
11 from tools import mlir_pytaco_api as pt unknown
17 compressed = pt.compressed
18 dense = pt.dense
21 csf = pt.format([compressed, compressed, compressed])
22 rm = pt.format([dense, dense])
29 B = pt.read(os.path.join(_SCRIPT_PATH, "data/nell-2.tns"), csf)
33 C = pt.from_array(np.full((B.shape[1], 25), 1, dtype=np.float32))
34 D = pt.from_array(np.full((B.shape[2], 25), 2, dtype=np.float32))
37 A = pt.tensor([B.shape[0], 25], rm)
40 i, j, k, l = pt.get_index_vars(4)
51 pt.write(out_file, A)