Searched refs:stack_node (Results 1 – 9 of 9) sorted by relevance
2119 stack_node: torch.fx.Node,2123 stack_shape = stack_node.meta["example_value"].shape2124 stack_dim = _get_dim(stack_node)2160 stack_node.replace_all_uses_with(reshape_node)2161 reshape_node.meta.update(stack_node.meta)2162 stack_inputs = stack_node.args[0] # type: ignore[union-attr]2164 graph.erase_node(stack_node)2172 stack_node: torch.fx.Node,2246 for stack_node in next_users:2247 if stack_node.target != torch.stack or not is_node_meta_valid(stack_node):[all …]
487 const NodeDef* stack_node = graph_view.GetNode(fanout_node.input(0)); in GetStackPushNodesToConvert() local488 while (stack_node->op() != "Stack" && stack_node->op() != "StackV2" && in GetStackPushNodesToConvert()489 stack_node->input_size() > 0 && in GetStackPushNodesToConvert()490 graph_view.HasNode(stack_node->input(0))) { in GetStackPushNodesToConvert()491 stack_node = graph_view.GetNode(stack_node->input(0)); in GetStackPushNodesToConvert()493 if (nodes_to_preserve.find(stack_node->name()) == in GetStackPushNodesToConvert()
640 stack_node = _node_def_pb2.NodeDef()641 stack_node.op = "Unpack"642 stack_node.name = "OpHintUnstack-%s" % flattened[0].name643 stack_node.attr["num"].i = len(flattened)645 stack_node.attr["T"].type = output_type646 stack_node.input.append(648 out_graphdef.node.extend([stack_node])653 output_node.input.append(_tensorflow_output_name(stack_node.name, idx))
113 stack_node = pop()115 last, node = next(stack_node)124 push(stack_node)
34808 struct list_head stack_node; member