Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/jit/passes/
Dtensorexpr_fuser.cpp1374 bool allow_stack_outputs = false; in createTensorExprOp() local
1375 if (node->hasAttribute(attr::allow_stack_outputs)) { in createTensorExprOp()
1376 allow_stack_outputs = node->i(attr::allow_stack_outputs) == 1; in createTensorExprOp()
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()
Dsymbolic_shape_runtime_fusion.cpp465 guarded_node->i_(attr::allow_stack_outputs, 1); in insertDynamicShapesGuard()
/external/pytorch/aten/src/ATen/core/
Dinterned_strings.h308 _(attr, allow_stack_outputs) \