Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/jit/passes/
Dlower_tuples.cpp41 Node* block_node = n; in flattenTupleInLoopParams() local
49 block_node->insertInput(index + j + 1, input->node()->inputs().at(j)); in flattenTupleInLoopParams()
56 block_node->removeInput(index); in flattenTupleInLoopParams()
64 Node* block_node = block->owningNode(); in flattenTupleInBlockReturn() local
75 if (block_node == nullptr) in flattenTupleInBlockReturn()
81 if (block_node->kind() == prim::Loop) in flattenTupleInBlockReturn()
83 auto tuple_output = block_node->outputs().at(index); in flattenTupleInBlockReturn()
90 new_construct_node->insertAfter(block_node); in flattenTupleInBlockReturn()
92 auto new_block_out = block_node->insertOutput(index + j + 1); in flattenTupleInBlockReturn()
97 new_construct_node->copyMetadata(block_node); in flattenTupleInBlockReturn()
[all …]
Dguard_elimination.cpp75 auto block_node = *it++; in moveGuardsToDefs() local
76 if (block_node->kind() != prim::Guard) { in moveGuardsToDefs()
79 block_node->moveBefore(b->owningNode()); in moveGuardsToDefs()
Dspecialize_autogradzero.cpp416 auto block_node = *it++; in specializeAutogradOps() local
417 block_node->moveBefore(n); in specializeAutogradOps()
/external/pytorch/torch/csrc/jit/frontend/
Dinline_loop_condition.cpp13 auto block_node = *it++; in InlineBlockBeforeNode() local
14 block_node->moveBefore(before_node); in InlineBlockBeforeNode()
/external/angle/build/gn_ast/
Dgn_ast.py327 func_params_node, block_node = children
328 if block_node.get(NODE_TYPE) != 'BLOCK':
/external/cronet/stable/build/gn_ast/
Dgn_ast.py327 func_params_node, block_node = children
328 if block_node.get(NODE_TYPE) != 'BLOCK':
/external/cronet/tot/build/gn_ast/
Dgn_ast.py327 func_params_node, block_node = children
328 if block_node.get(NODE_TYPE) != 'BLOCK':
/external/selinux/libsepol/cil/src/
Dcil_resolve_ast.c2320 struct cil_tree_node *block_node = NULL; in cil_resolve_blockabstract() local
2328 block_node = NODE(block_datum); in cil_resolve_blockabstract()
2329 if (block_node->flavor != CIL_BLOCK) { in cil_resolve_blockabstract()
2337 cil_list_append(abstract_blocks, CIL_NODE, block_node); in cil_resolve_blockabstract()
2349 struct cil_tree_node *block_node = NULL; in cil_resolve_in() local
2359 block_node = NODE(block_datum); in cil_resolve_in()
2361 if (block_node->flavor == CIL_OPTIONAL) { in cil_resolve_in()
2364 cil_tree_log(block_node, CIL_ERR, "First optional block"); in cil_resolve_in()
2370 rc = cil_copy_ast(db, current, block_node); in cil_resolve_in()
3456 struct cil_tree_node *block_node = NODE(bi->block); in cil_check_inheritances() local
[all …]
/external/pytorch/torch/_export/
Dconverter.py499 for block_node in block.nodes():
500 for block_node_in in block_node.inputs():
507 self._identify_inputs_as_arguments(block_node)
/external/pytorch/torch/csrc/jit/ir/
Dir.cpp661 Node* block_node = insertNode(create(prim::TracedModuleForward, 0)); in push_scope() local
662 block_node->s_(attr::scope, scope_name); in push_scope()
663 Block* b = block_node->addBlock(); in push_scope()