Home
last modified time | relevance | path

Searched refs:inp_list (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/backends/xnnpack/
H A Dxnnpack_backend_preprocess.cpp72 c10::List<at::Tensor> inp_list = inp.toTensorList(); in preprocess() local
74 graph->inputs().size() == inp_list.size(), in preprocess()
77 example_inputs.reserve(inp_list.size()); in preprocess()
78 for (const auto i : c10::irange(inp_list.size())) { in preprocess()
79 example_inputs.emplace_back(inp_list[i]); in preprocess()
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_cudagraph_trees.py959 inp_list = [inp]
960 foo_cg = self.cudagraphify_impl(foo, inp_list, ())
961 foo_cg(inp_list)
1425 inp_list = [inp]
1426 foo_cg = self.cudagraphify_impl(foo, inp_list, ())
1427 x1, x2 = foo_cg(inp_list)
1434 inp_list = [x1]
1435 foo2_cg = self.cudagraphify_impl(foo2, inp_list, ())
1436 foo2_cg(inp_list)
1459 inp_list = [inp]
[all …]
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dconcat_opt.cpp25 auto inp_list = n->input(0); in removeCatNodeFromGraph() local
28 if (!inp_list->hasUses()) { in removeCatNodeFromGraph()
29 GRAPH_UPDATE("Deleting\n", *inp_list->node()); in removeCatNodeFromGraph()
30 inp_list->node()->destroy(); in removeCatNodeFromGraph()
/aosp_15_r20/external/pytorch/torch/_export/
H A Dconverter.py649 inp_list = [
654 tuple(inp_list),
737 inp_list = [
742 tuple(inp_list),
/aosp_15_r20/external/pytorch/test/nn/
H A Dtest_embedding.py356 inp_list = [0, 1, 2, 2]
357 inp = torch.tensor(inp_list, device=device)
/aosp_15_r20/external/pytorch/torch/csrc/jit/frontend/
H A Dir_emitter.cpp3578 auto inp_list = in emitApplySpecialForm() local
3583 {inp_list}, in emitApplySpecialForm()
3586 auto inp_list = in emitApplySpecialForm() local
3591 {inp_list}, in emitApplySpecialForm()
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_repros.py3536 def fn(x, inp_list): argument
3537 d = DClass(inp_list)