Searched refs:stack_node (Results 1 – 2 of 2) sorted by relevance
488 const NodeDef* stack_node = graph_view.GetNode(fanout_node.input(0)); in GetStackPushNodesToConvert() local489 while (stack_node->op() != "Stack" && stack_node->op() != "StackV2" && in GetStackPushNodesToConvert()490 stack_node->input_size() > 0 && in GetStackPushNodesToConvert()491 graph_view.HasNode(stack_node->input(0))) { in GetStackPushNodesToConvert()492 stack_node = graph_view.GetNode(stack_node->input(0)); in GetStackPushNodesToConvert()494 if (nodes_to_preserve.find(stack_node->name()) == in GetStackPushNodesToConvert()
632 stack_node = _node_def_pb2.NodeDef()633 stack_node.op = "Unpack"634 stack_node.name = "OpHintUnstack-%s" % flattened[0].name635 stack_node.attr["num"].i = len(flattened)637 stack_node.attr["T"].type = output_type638 stack_node.input.append(_tensorflow_output_name(640 out_graphdef.node.extend([stack_node])645 output_node.input.append(_tensorflow_output_name(stack_node.name, idx))