Searched refs:num_leaves (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | pytree.h | 129 int num_leaves() const { in num_leaves() function
133 return traversal_.back().num_leaves; in num_leaves()
170 int num_leaves = 0; member
|
| D | pytree.cc | 172 node.num_leaves = leaves.size() - start_num_leaves; in FlattenInto()
204 "Too few leaves for PyTreeDef; expected %d, got %d", num_leaves(), in Unflatten()
232 "Too many leaves for PyTreeDef; expected %d.", num_leaves())); in Unflatten()
294 py::list leaves(num_leaves()); in FlattenUpTo()
298 int leaf = num_leaves() - 1; in FlattenUpTo()
532 out_root.num_nodes = (root.num_nodes - root.num_leaves) + in Compose()
533 (inner_root.num_nodes * root.num_leaves); in Compose()
534 out_root.num_leaves *= inner_root.num_leaves; in Compose()
641 .def_property_readonly("num_leaves", &PyTreeDef::num_leaves) in BuildPytreeSubmodule()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition_test.cc | 560 const int num_leaves = 20; in TEST() local
561 for (int i = 0; i < num_leaves; ++i) { in TEST()
562 indexes.push_back((i + 2001) % num_leaves); in TEST()
607 int square_index = num_leaves - 1 - i; in TEST()
626 const int num_leaves = 20; in TEST() local
627 for (int i = 0; i < num_leaves; ++i) { in TEST()
628 indexes.push_back((i + 2001) % num_leaves); in TEST()
702 int square_index = num_leaves - 1 - i; in TEST()
|
| /external/webrtc/modules/audio_processing/transient/ |
| D | wpd_tree.h | 81 int num_leaves() const { return 1 << levels_; } in num_leaves() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_loop_simplifier.cc | 813 int64 num_leaves = 0; in UnflattenTupleInstr() local
817 ++num_leaves; in UnflattenTupleInstr()
822 UnflattenTupleInstr(instrs.subspan(0, num_leaves), in UnflattenTupleInstr()
826 instrs.remove_prefix(num_leaves); in UnflattenTupleInstr()
|
| /external/icing/icing/legacy/index/ |
| D | icing-dynamic-trie.h | 157 uint32_t num_leaves; member
|
| D | icing-dynamic-trie.cc | 1370 stats->num_leaves++; in CollectStatsRecursive()
1471 total_wasted, num_intermediates, num_leaves, suffixes_used, in DumpStats()
|