HomeSort by relevance Sort by last modified time
    Searched refs:stack_node (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/grappler/optimizers/
loop_optimizer.cc 488 const NodeDef* stack_node = graph_view.GetNode(fanout_node.input(0)); local
489 while (stack_node->op() != "Stack" && stack_node->op() != "StackV2" &&
490 stack_node->input_size() > 0 &&
491 graph_view.HasNode(stack_node->input(0))) {
492 stack_node = graph_view.GetNode(stack_node->input(0));
494 if (nodes_to_preserve.find(stack_node->name()) ==
    [all...]
  /external/tensorflow/tensorflow/lite/python/
op_hint.py 632 stack_node = _node_def_pb2.NodeDef()
633 stack_node.op = "Unpack"
634 stack_node.name = "OpHintUnstack-%s" % flattened[0].name
635 stack_node.attr["num"].i = len(flattened)
637 stack_node.attr["T"].type = output_type
638 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))
    [all...]

Completed in 257 milliseconds