Searched refs:TensorCreationOpsModule (Results 1 – 2 of 2) sorted by relevance
35 TensorCreationOpsModule,69 "tensor_creation_ops": TensorCreationOpsModule(),
101 class TensorCreationOpsModule(torch.nn.Module): class