Searched refs:from_node_name (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/ |
| D | mutable_graph_view.cc | 116 absl::string_view from_node_name(control_fanouts->first.node->name()); in SwapControlledFanoutInputs() local
123 if (tensor_id.node() == from_node_name) { in SwapControlledFanoutInputs()
257 ErrorHandler UpdateFanoutsError(absl::string_view from_node_name, in UpdateFanoutsError() argument
259 return [from_node_name, to_node_name](absl::string_view msg) { in UpdateFanoutsError()
261 from_node_name, to_node_name); in UpdateFanoutsError()
547 Status MutableGraphView::UpdateNodeName(absl::string_view from_node_name, in UpdateNodeName() argument
550 auto error_status = [from_node_name, to_node_name, in UpdateNodeName()
554 from_node_name, to_node_name, update_fanouts); in UpdateNodeName()
558 NodeDef* node = GetNode(from_node_name); in UpdateNodeName()
559 TF_RETURN_IF_ERROR(CheckNodeExists(from_node_name, node, error_status)); in UpdateNodeName()
[all …]
|
| D | mutable_graph_view.h | 91 Status UpdateNodeName(absl::string_view from_node_name,
118 Status SwapNodeNames(absl::string_view from_node_name,
133 Status UpdateFanouts(absl::string_view from_node_name,
|
| D | mutable_graph_view_test.cc | 338 void TestUpdateNodeName(absl::string_view from_node_name, bool node_exists, in TestUpdateNodeName() argument
349 NodeDef* node = graph.GetNode(from_node_name); in TestUpdateNodeName()
357 GetNodeInputsFromGraph(graph_def, from_node_name); in TestUpdateNodeName()
359 Status s = graph.UpdateNodeName(from_node_name, to_node_name, update_fanouts); in TestUpdateNodeName()
365 updated_node_name = string(from_node_name); in TestUpdateNodeName()
674 void TestSwapNodeNamesError(absl::string_view from_node_name, in TestSwapNodeNamesError() argument
681 Status s = graph.SwapNodeNames(from_node_name, to_node_name, update_fanouts); in TestSwapNodeNamesError()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | scoped_allocator_optimizer.cc | 179 Status RemoveEdge(const string& input_edge_name, const string& from_node_name, in RemoveEdge() argument
182 node_map->RemoveOutput(from_node_name, to_node->name()); in RemoveEdge()
|