Searched refs:DiagonalTensor (Results 1 – 1 of 1) sorted by relevance
91 class DiagonalTensor: class259 class SubDiagonalTensor(DiagonalTensor):384 torch._dynamo.config.traceable_tensor_subclasses.add(DiagonalTensor)399 t1 = DiagonalTensor(5, 2)415 t1 = DiagonalTensor(5, 2)464 t1 = DiagonalTensor(5, 2)533 t1 = DiagonalTensor(5, 2)534 t2 = DiagonalTensor(5, 2)