Searched defs:orig_outputs (Results 1 – 5 of 5) sorted by relevance
/aosp_15_r20/external/pytorch/test/cpp/jit/ |
H A D | test_concat_opt.cpp | 33 auto orig_outputs = runGraph(graph, inputs); in TEST() local 78 auto orig_outputs = runGraph(graph, inputs); in TEST() local 129 auto orig_outputs = runGraph(graph, inputs); in TEST() local 167 auto orig_outputs = runGraph(graph, inputs); in TEST() local 201 auto orig_outputs = runGraph(graph, inputs); in TEST() local 257 auto orig_outputs = runGraph(graph, inputs); in TEST() local 292 auto orig_outputs = runGraph(graph, inputs); in TEST() local 329 auto orig_outputs = runGraph(graph, inputs); in TEST() local 374 auto orig_outputs = runGraph(graph, inputs); in TEST() local 414 auto orig_outputs = runGraph(graph, inputs); in TEST() local [all …]
|
H A D | test_stack_opt.cpp | 37 auto orig_outputs = runGraph(graph, inputs); in TEST() local 82 auto orig_outputs = runGraph(graph, inputs); in TEST() local 122 auto orig_outputs = runGraph(graph, inputs); in TEST() local 165 auto orig_outputs = runGraph(graph, inputs); in TEST() local 210 auto orig_outputs = runGraph(graph, inputs); in TEST() local 279 auto orig_outputs = runGraph(graph, inputs); in TEST() local
|
/aosp_15_r20/external/pytorch/torch/fx/passes/ |
H A D | split_utils.py | 53 orig_outputs: List = field(default_factory=list) variable in Component
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
H A D | canonicalize_graph_fuser_ops.cpp | 72 if (auto orig_outputs = getChunkOutputs(*it)) { in CanonicalizeOps() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
H A D | split_utils_test.cc | 157 std::vector<Tensor> orig_outputs; in CheckSplitFunctions() local
|