Home
last modified time | relevance | path

Searched refs:num_leaves (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/python/layers/
Ddecisions_to_data.py69 [0, self.params.num_nodes - self.params.num_leaves - 1],
70 [-1, self.params.num_leaves])
114 [0, self.params.num_nodes - self.params.num_leaves - 1],
115 [-1, self.params.num_leaves])
152 [0, self.params.num_nodes - self.params.num_leaves - 1],
153 [-1, self.params.num_leaves])
188 [0, self.params.num_nodes - self.params.num_leaves - 1],
189 [-1, self.params.num_leaves])
204 [0, self.params.num_nodes - self.params.num_leaves - 1],
205 [-1, self.params.num_leaves])
[all …]
Ddecisions_to_data_test.py51 self.params.num_leaves = 2**(self.params.hybrid_tree_depth - 1)
/external/tensorflow/tensorflow/core/graph/
Dgraph_partition_test.cc560 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/tensorflow/tensorflow/contrib/tensor_forest/kernels/
Dstats_ops.cc292 const int32 num_leaves = leaf_examples.size(); in Compute() local
323 num_leaves](int64 start, int64 end) { in Compute()
325 CHECK(end <= num_leaves); in Compute()
333 Shard(num_threads, worker_threads->workers, num_leaves, costPerUpdate, in Compute()
/external/webrtc/webrtc/modules/audio_processing/transient/
Dwpd_tree.h80 int num_leaves() const { return 1 << levels_; } in num_leaves() function
/external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/python/
Dhybrid_layer_test.py45 self.params.num_leaves = 2**(self.params.hybrid_tree_depth - 1)
/external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/python/models/
Dk_feature_decisions_to_data_then_nn_test.py54 self.params.num_leaves = 2**(self.params.hybrid_tree_depth - 1)
Dforest_to_data_then_nn_test.py54 self.params.num_leaves = 2**(self.params.hybrid_tree_depth - 1)
Ddecisions_to_data_then_nn_test.py52 self.params.num_leaves = 2**(self.params.hybrid_tree_depth - 1)
/external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/python/kernel_tests/
Dk_feature_routing_function_op_test.py50 self.params.num_leaves = 2**(self.params.hybrid_tree_depth - 1)
/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_simplifier.cc601 int64 num_leaves = 0; in UnflattenTupleInstr() local
605 ++num_leaves; in UnflattenTupleInstr()
610 UnflattenTupleInstr(instrs.subspan(0, num_leaves), in UnflattenTupleInstr()
614 instrs.remove_prefix(num_leaves); in UnflattenTupleInstr()