Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/graph/
Dgraph_partition_test.cc214 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::__anonc404fe730111::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/
Dmaster_session.cc106 for (Part& part : partitions_) { in ~ReffedClientGraph()
135 for (auto& p : partitions_) { in SetRPCLogging()
166 int waiting_for = partitions_.size(); in RetrieveLogs()
170 for (auto& p : partitions_) { in RetrieveLogs()
276 std::vector<Part> partitions_; member in tensorflow::MasterSession::ReffedClientGraph
356 graph_defs_for_publishing.reserve(partitions_.size()); in RegisterPartitions()
443 partitions_.reserve(graph_partitions.size()); in DoRegisterPartitions()
446 partitions_.emplace_back(); in DoRegisterPartitions()
447 Part* part = &partitions_.back(); in DoRegisterPartitions()
457 for (Part& part : partitions_) { in DoRegisterPartitions()
[all …]
/external/tensorflow/tensorflow/lite/delegates/
Dutils.h92 int num_partitions() const { return partitions_.size(); } in num_partitions()
109 std::vector<TfLiteDelegateParams*> partitions_; variable
Dutils.cc65 partitions_.push_back(partition_params_array_ + i); in Partition()
78 std::vector<TfLiteDelegateParams*> sorted_partitions(partitions_); in GetFirstNLargestPartitions()