Home
last modified time | relevance | path

Searched refs:SafeTensorIdToString (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/
H A Dutils_test.cc554 TEST_F(UtilsTest, SafeTensorIdToString) { in TEST_F() argument
555 EXPECT_EQ(SafeTensorIdToString({"foo", -1}), "^foo"); in TEST_F()
556 EXPECT_EQ(SafeTensorIdToString({"foo", 0}), "foo"); in TEST_F()
557 EXPECT_EQ(SafeTensorIdToString({"foo", 1}), "foo:1"); in TEST_F()
558 EXPECT_EQ(SafeTensorIdToString({"foo", 2}), "foo:2"); in TEST_F()
H A Dutils.h298 string SafeTensorIdToString(const SafeTensorId& tensor_id);
H A Dutils.cc81 string SafeTensorIdToString(const SafeTensorId& tensor_id) { in SafeTensorIdToString() function
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/
H A Dgraph_view.cc1235 node_def->add_input(SafeTensorIdToString(fanin)); in ApplyNodeUpdates()
1260 SafeTensorIdToString(update_fanin.second)); in ApplyNodeUpdates()
1284 new_node_def->add_input(SafeTensorIdToString(fanin)); in SetNewNodesFanins()