Searched refs:tuple_node (Results 1 – 6 of 6) sorted by relevance
74 const xrt::XLATupleNode& tuple_node, const OpInputList& input_tensor_list, in ParseTupleNode() argument77 if (tuple_node.tuples_size() > 0) { in ParseTupleNode()80 std::vector<xla::Shape> subshapes(tuple_node.tuples_size(), dummy); in ParseTupleNode()83 for (int i = 0; i < tuple_node.tuples_size(); ++i) { in ParseTupleNode()85 tuple_node.tuples(i), input_tensor_list, input_vector, in ParseTupleNode()90 int input_index = tuple_node.input_index(); in ParseTupleNode()96 bool release_this_input = tuple_node.release_input_handle(); in ParseTupleNode()146 const xrt::XLATupleNode* tuple_node = &tuple_tree_root; in ParseTupleTree()148 tuple_node = &tuple_node->tuples(index[i]); in ParseTupleTree()152 int input_index = tuple_node->input_index(); in ParseTupleTree()[all …]
420 auto tuple_node = graph->createTuple(out_node->inputs()); in replaceListOutputWithTuple() local421 tuple_node->insertAfter(out_node); in replaceListOutputWithTuple()422 out->replaceAllUsesWith(tuple_node->output()); in replaceListOutputWithTuple()
234 auto tuple_node = graph->outputs().at(0)->node(); in transformShapeFunction() local236 auto tuple_unpack_values = createTupleUnpack(tuple_node->output()); in transformShapeFunction()
279 auto tuple_node = graph->createTuple( in getOutput() local281 graph->insertNode(tuple_node); in getOutput()282 return tuple_node->output(); in getOutput()
169 tuple_node = tuple_value.node()176 return tuple(tuple_node.inputs())186 tuple_node = tuple_value.node()195 unpacked = tuple(tuple_node.inputs())
837 auto tuple_node = *node_iter; in TEST() local841 EXPECT_TRUE(aliasDb.mayContainAlias(input, tuple_node->output())); in TEST()