Searched refs:placeholder_node (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/tools/ |
| D | strip_unused_lib.py | 64 placeholder_node = node_def_pb2.NodeDef()
65 placeholder_node.op = "Placeholder"
66 placeholder_node.name = node.name
69 placeholder_node.attr["dtype"].CopyFrom(
73 placeholder_node.attr["dtype"].CopyFrom(
76 placeholder_node.attr["_output_shapes"].CopyFrom(node.attr[
79 placeholder_node.attr["shape"].CopyFrom(node.attr["shape"])
80 inputs_replaced_graph_def.node.extend([placeholder_node])
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | strip_unused_nodes.cc | 170 NodeDef placeholder_node; in StripUnusedNodes() local
172 placeholder_node = node; in StripUnusedNodes()
174 placeholder_node.set_op("Placeholder"); in StripUnusedNodes()
175 placeholder_node.set_name(node.name()); in StripUnusedNodes()
180 SetNodeAttr("dtype", type, &placeholder_node); in StripUnusedNodes()
181 SetNodeAttr("shape", shape, &placeholder_node); in StripUnusedNodes()
183 *(output_graph_def->mutable_node()->Add()) = placeholder_node; in StripUnusedNodes()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_util.cc | 155 Node* placeholder_node; in PreprocessDataEdgesBetweenOutsideCompilations() local
172 placeholder_node = g->AddNode(placeholder_def, &s); in PreprocessDataEdgesBetweenOutsideCompilations()
174 placeholders[placeholder_index] = placeholder_node; in PreprocessDataEdgesBetweenOutsideCompilations()
176 placeholder_node = iter->second; in PreprocessDataEdgesBetweenOutsideCompilations()
178 g->AddEdge(placeholder_node, 0, dst, dst_input); in PreprocessDataEdgesBetweenOutsideCompilations()
182 *new_def.mutable_input(dst_input) = placeholder_node->name(); in PreprocessDataEdgesBetweenOutsideCompilations()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_while_op_test.cc | 207 const Node* placeholder_node = nullptr; in TEST() local
210 placeholder_node = op; in TEST()
213 ASSERT_NE(placeholder_node, nullptr); in TEST()
217 for (const Node* consumer : placeholder_node->out_nodes()) { in TEST()
328 const Node* placeholder_node = nullptr; in TEST() local
331 placeholder_node = op; in TEST()
334 ASSERT_NE(placeholder_node, nullptr); in TEST()
338 for (const Node* consumer : placeholder_node->out_nodes()) { in TEST()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 623 Node* placeholder_node; in CreatePlaceholderNodeForFeed() local
627 TF_RETURN_IF_ERROR(builder.Finalize(graph_.get(), &placeholder_node)); in CreatePlaceholderNodeForFeed()
641 TF_RETURN_IF_ERROR(graph_->UpdateEdge(placeholder_node, 0, edge->dst(), in CreatePlaceholderNodeForFeed()
653 graph_->AddControlEdge(placeholder_node, edge->dst()); in CreatePlaceholderNodeForFeed()
661 return std::pair<Node*, bool>(placeholder_node, update_inplace); in CreatePlaceholderNodeForFeed()
751 Node* placeholder_node = placeholder_node_and_removed.first; in AddNodesToShapeRefiner() local
754 node = placeholder_node; in AddNodesToShapeRefiner()
757 remapped_feeds_[{it->first, index}] = placeholder_node->name(); in AddNodesToShapeRefiner()
758 (*node_name_map)[placeholder_node->name()] = placeholder_node; in AddNodesToShapeRefiner()
760 Status status = shape_refiner_->AddNode(placeholder_node); in AddNodesToShapeRefiner()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_execute_utils.cc | 1446 NodeDef placeholder_node; in ReplaceInputNodeByPlaceHolder() local
1447 placeholder_node.set_op("Placeholder"); in ReplaceInputNodeByPlaceHolder()
1448 placeholder_node.set_name(node_name); in ReplaceInputNodeByPlaceHolder()
1449 AddNodeAttr("dtype", type, &placeholder_node); in ReplaceInputNodeByPlaceHolder()
1450 AddNodeAttr("shape", shape, &placeholder_node); in ReplaceInputNodeByPlaceHolder()
1452 AddOutputTensorShapeType({type}, {shape}, &placeholder_node); in ReplaceInputNodeByPlaceHolder()
1454 node = placeholder_node; in ReplaceInputNodeByPlaceHolder()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 1556 Node* placeholder_node = xla_graph->AddNode(placeholder_def, &s); in RemoveEdgesBetweenArgAndRetval() local
1562 xla_graph->AddEdge(placeholder_node, 0, dst, dst_input); in RemoveEdgesBetweenArgAndRetval()
|