Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/jit/passes/
Dtensorexpr_fuser.cpp1421 auto num_subgraph_inputs = subgraph->inputs().size(); in createTensorExprOp() local
1422 return [kernel, num_subgraph_inputs, allow_stack_outputs](Stack& stack) { in createTensorExprOp()
1432 if (num_subgraph_inputs == stack.size() || !allow_stack_outputs) { in createTensorExprOp()