Searched refs:TensorIdToString (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | graph_compiler_util.cc | 63 const string key = TensorIdToString(feed.id()); in AddArgNodes()
87 .Attr(kFeedIdAttr, TensorIdToString(feed.id())) in AddArgNodes()
121 return errors::NotFound("Can't find fetch id: ", TensorIdToString(id)); in AddRetvalNodes()
125 return errors::InvalidArgument("Invalid fetch id: ", TensorIdToString(id), in AddRetvalNodes()
137 .Attr(kFetchIdAttr, TensorIdToString(id)) in AddRetvalNodes()
167 missing_feeds.insert(TensorIdToString(feed.id())); in RewriteAndPruneGraph()
170 missing_fetches.insert(TensorIdToString(fetch.id())); in RewriteAndPruneGraph()
|
| D | tf2xla_util.h | 52 string TensorIdToString(const tf2xla::TensorId& id);
|
| D | mlir_tf2xla.cc | 66 feed_name_remap.at(TensorIdToString(feed.id())); in ConvertInputInfo()
|
| D | tf2xla_util.cc | 519 const string name_port = TensorIdToString(feed->id()); in AddPlaceholdersForFeeds()
540 TensorIdToString(feed_id)); in AddPlaceholdersForFeeds()
667 string TensorIdToString(const tf2xla::TensorId& id) { in TensorIdToString() function
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils_test.cc | 547 TEST_F(UtilsTest, TensorIdToString) { in TEST_F() argument
548 EXPECT_EQ(TensorIdToString({"foo", -1}), "^foo"); in TEST_F()
549 EXPECT_EQ(TensorIdToString({"foo", 0}), "foo"); in TEST_F()
550 EXPECT_EQ(TensorIdToString({"foo", 1}), "foo:1"); in TEST_F()
551 EXPECT_EQ(TensorIdToString({"foo", 2}), "foo:2"); in TEST_F()
|
| D | mutable_graph_view.cc | 122 string control = TensorIdToString({to_node_name, Graph::kControlSlot}); in SwapControlledFanoutInputs()
146 string input = TensorIdToString({to_node_name, i}); in SwapRegularFanoutInputs()
691 node->set_input(i, TensorIdToString({old_node_name, idx})); in SwapNodeNames()
841 TensorIdToString({to_node->name(), output_port.port_id})); in UpdateFanoutsInternal()
889 node->add_input(TensorIdToString({fanin.node->name(), fanin.port_id})); in AddFaninInternal()
955 node->add_input(TensorIdToString(fanin)); in AddRegularFaninByPort()
1031 new_node.add_input(TensorIdToString({fanin.node->name(), fanin.port_id})); in GetOrCreateIdentityConsumingSwitch()
1312 string to_fanin_string = TensorIdToString(to_fanin); in UpdateFanin()
1382 node->set_input(port, TensorIdToString(fanin)); in UpdateRegularFaninByPort()
|
| D | utils.h | 294 string TensorIdToString(const TensorId& tensor_id);
|
| D | utils.cc | 76 string TensorIdToString(const TensorId& tensor_id) { in TensorIdToString() function
|
| D | mutable_graph_view_test.cc | 1682 string tensor_id_str = TensorIdToString(fanin); in TestUpdateFaninFromFaninToNodeAsSwitchControl()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 1432 node.set_input(idx, TensorIdToString(mapping->second)); in RestoreTensorMapping()
|
| D | generic_layout_optimizer_transposer.cc | 1308 TensorIdToString({fanin_node->GetName(), fanin.index()}), in MaybeReshapeVectorFanin()
|