Searched refs:TensorIdToString (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla.cc | 74 const string key = TensorIdToString(feed.id()); in AddArgNodes()
96 .Attr(kFeedIdAttr, TensorIdToString(feed.id())) in AddArgNodes()
129 return errors::NotFound("Can't find fetch id: ", TensorIdToString(id)); in AddRetvalNodes()
133 return errors::InvalidArgument("Invalid fetch id: ", TensorIdToString(id), in AddRetvalNodes()
144 .Attr(kFetchIdAttr, TensorIdToString(id)) in AddRetvalNodes()
174 missing_feeds.insert(TensorIdToString(feed.id())); in RewriteAndPruneGraph()
177 missing_fetches.insert(TensorIdToString(fetch.id())); in RewriteAndPruneGraph()
|
| D | tf2xla_util.h | 50 string TensorIdToString(const tf2xla::TensorId& id);
|
| D | tf2xla_util.cc | 346 const string name_port = TensorIdToString(feed->id()); in AddPlaceholdersForFeeds()
367 TensorIdToString(feed_id)); in AddPlaceholdersForFeeds()
496 string TensorIdToString(const tf2xla::TensorId& id) { in TensorIdToString() function
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils_test.cc | 470 TEST_F(UtilsTest, TensorIdToString) { in TEST_F() argument
471 EXPECT_EQ(TensorIdToString({"foo", -1}), "^foo"); in TEST_F()
472 EXPECT_EQ(TensorIdToString({"foo", 0}), "foo"); in TEST_F()
473 EXPECT_EQ(TensorIdToString({"foo", 1}), "foo:1"); in TEST_F()
474 EXPECT_EQ(TensorIdToString({"foo", 2}), "foo:2"); in TEST_F()
|
| D | mutable_graph_view.cc | 117 string control = TensorIdToString({to_node_name, Graph::kControlSlot}); in SwapControlledFanoutInputs()
141 string input = TensorIdToString({to_node_name, i}); in SwapRegularFanoutInputs()
669 node->set_input(i, TensorIdToString({old_node_name, idx})); in SwapNodeNames()
816 TensorIdToString({to_node->name(), output_port.port_id})); in UpdateFanoutsInternal()
864 node->add_input(TensorIdToString({fanin.node->name(), fanin.port_id})); in AddFaninInternal()
930 node->add_input(TensorIdToString(fanin)); in AddRegularFaninByPort()
1006 new_node.add_input(TensorIdToString({fanin.node->name(), fanin.port_id})); in GetOrCreateIdentityConsumingSwitch()
1287 string to_fanin_string = TensorIdToString(to_fanin); in UpdateFanin()
1363 node->set_input(port, TensorIdToString(fanin)); in UpdateRegularFaninByPort()
|
| D | utils.h | 105 string TensorIdToString(const TensorId& tensor_id);
|
| D | utils.cc | 160 string TensorIdToString(const TensorId& tensor_id) { in TensorIdToString() function
|
| D | mutable_graph_view_test.cc | 1682 string tensor_id_str = TensorIdToString(fanin); in TestUpdateFaninFromFaninToNodeAsSwitchControl()
|