Searched refs:tuple_output (Results 1 – 3 of 3) sorted by relevance
68 tuple_output = (output,)70 tuple_output = output # type: ignore[assignment]71 assert len(output_differentiability) == len(tuple_output)73 for idx, (differentiable, out) in enumerate(zip(output_differentiability, tuple_output)):
83 auto tuple_output = block_node->outputs().at(index); in flattenTupleInBlockReturn() local86 if (!(tuple_output->type()->cast<TupleType>())) in flattenTupleInBlockReturn()96 new_construct_node->output()->setType(tuple_output->type()); in flattenTupleInBlockReturn()98 tuple_output->replaceAllUsesWith(new_construct_node->output()); in flattenTupleInBlockReturn()
866 tuple_output = list(ge.graph.outputs())[0]867 tuple_inputs = list(tuple_output.node().inputs())