Searched refs:unpack_node (Results 1 – 2 of 2) sorted by relevance
358 auto unpack_node = in addInput() local360 auto elem_values = unpack_node->outputs(); in addInput()566 Node* unpack_node = in setValue() local569 setValue(outputs.get(i), unpack_node->outputs()[i]); in setValue()573 Node* unpack_node = graph->insertNode(graph->createTupleUnpack(value)); in setValue() local575 setValue(outputs[i], unpack_node->outputs()[i]); in setValue()579 Node* unpack_node = in setValue() local582 setValue(elements[i], unpack_node->outputs()[i]); in setValue()898 Node* unpack_node = graph->insertNode( in addOutput() local901 Value* output_val = unpack_node->outputs()[i]; in addOutput()
595 auto* unpack_node = g->insertNode( in broadcast_tensors() local603 TORCH_INTERNAL_ASSERT(unpack_node->outputs().size() == inputs.size()); in broadcast_tensors()606 Value* broadcasted_output = unpack_node->outputs()[i]; in broadcast_tensors()610 return unpack_node->outputs(); in broadcast_tensors()