Searched refs:partitions_ (Results 1 – 4 of 4) 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::__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/ |
D | master_session.cc | 106 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/ |
D | utils.h | 92 int num_partitions() const { return partitions_.size(); } in num_partitions() 109 std::vector<TfLiteDelegateParams*> partitions_; variable
|
D | utils.cc | 65 partitions_.push_back(partition_params_array_ + i); in Partition() 78 std::vector<TfLiteDelegateParams*> sorted_partitions(partitions_); in GetFirstNLargestPartitions()
|