Searched refs:partitions_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition_test.cc | 214 TF_EXPECT_GRAPH_EQ(graph_def, partitions_[a]); in ExpectMatchA()
221 TF_EXPECT_GRAPH_EQ(graph_def, partitions_[b]); in ExpectMatchB()
237 std::unordered_map<string, GraphDef> partitions_; member in tensorflow::__anon8c703d150111::GraphPartitionTest
244 Partition(ToGraphDef(), &partitions_); in TEST_F()
245 EXPECT_EQ(1, partitions_.size()); in TEST_F()
257 Partition(ToGraphDef(), &partitions_); in TEST_F()
258 EXPECT_EQ(2, partitions_.size()); in TEST_F()
278 Partition(ToGraphDef(), &partitions_); in TEST_F()
279 EXPECT_EQ(2, partitions_.size()); in TEST_F()
302 Partition(ToGraphDef(), &partitions_); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | master_session.cc | 102 for (Part& part : partitions_) { in ~ReffedClientGraph()
131 for (auto& p : partitions_) { in SetRPCLogging()
162 int waiting_for = partitions_.size(); in RetrieveLogs()
166 for (auto& p : partitions_) { in RetrieveLogs()
272 std::vector<Part> partitions_; member in tensorflow::MasterSession::ReffedClientGraph
352 graph_defs_for_publishing.reserve(partitions_.size()); in RegisterPartitions()
439 partitions_.reserve(graph_partitions.size()); in DoRegisterPartitions()
442 partitions_.emplace_back(); in DoRegisterPartitions()
443 Part* part = &partitions_.back(); in DoRegisterPartitions()
453 for (Part& part : partitions_) { in DoRegisterPartitions()
[all …]
|