Lines Matching full:ones

158             return torch.add(x, torch.ones(1))
166             torch.ones(1),
172         (torch.ones(1),),
233             executorch_module.run_method("forward", (torch.ones(1), torch.zeros(1)))[0],
234             torch.ones(1),
237             executorch_module.run_method("foo", (torch.ones(1),))[0],
238             torch.ones(1) + torch.ones(1),
292         ep = to_edge(torch.export.export(mul, (torch.ones(1),))).exported_program()
322             torch.ones(1), torch.ones(1)
339                 torch.ones(1), torch.ones(1)
341             torch.ones(1),  # x * y * x
347                 torch.ones(1), torch.ones(1)
397                 torch.ones(1), torch.ones(1)
399             torch.ones(1),  # x * y * x
404                 torch.ones(1),
406             torch.ones(1) + 1,  # x + 1
419             forward_program.module()(torch.ones(1), torch.ones(1)),
420             torch.ones(1) + 1,  # x * y + x
477             forward_program.module()(torch.ones(1), torch.ones(1)),
478             torch.ones(1) + 1,  # x * y + x
543         one = torch.ones(1, dtype=torch.float)
544         two = torch.ones(1, dtype=torch.int32)