Home
last modified time | relevance | path

Searched refs:pack_node (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dreplicate_per_replica_nodes.cc141 Node* pack_node = graph->AddNode(pack_def, &status); in ReplicateFromCompositeDeviceToRegularDevice() local
143 pack_node->set_assigned_device_name(dst->assigned_device_name()); in ReplicateFromCompositeDeviceToRegularDevice()
146 pack_node, i); in ReplicateFromCompositeDeviceToRegularDevice()
148 graph->AddEdge(pack_node, /*x=*/0, dst, edge->dst_input()); in ReplicateFromCompositeDeviceToRegularDevice()
/external/tensorflow/tensorflow/lite/python/
Dop_hint.py1105 pack_node = node_name
1135 pack_node = node_name
1145 if input_op in visited and input_op != pack_node:
1159 (end_input if stripped == pack_node else non_stripped)