Home
last modified time | relevance | path

Searched refs:nodes_size (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dtraining_ops.cc276 CHECK_EQ(tree_config->nodes_size(), 1); in RetrieveBias()
432 if (tree_config->nodes_size() <= 0) { in Compute()
725 CHECK(node_id < tree_config->nodes_size()) in SplitTreeNode()
735 int32 left_id = tree_config->nodes_size(); in SplitTreeNode()
741 int32 right_id = tree_config->nodes_size(); in SplitTreeNode()
773 while (depth < tree_config->nodes_size() && in SplitTreeLayer()
777 CHECK(tree_config->nodes_size() > 0) in SplitTreeLayer()
814 for (int idx = depth; idx < tree_config->nodes_size(); idx++) { in SplitTreeLayer()
821 int nodes_size = tree_config->nodes_size(); in SplitTreeLayer() local
823 if (idx + depth + 1 < nodes_size) { in SplitTreeLayer()
[all …]
/external/pdfium/third_party/libopenjpeg20/
Dtgt.c97 tree->nodes_size = tree->numnodes * (OPJ_UINT32)sizeof(opj_tgt_node_t); in opj_tgt_create()
178 if (l_node_size > p_tree->nodes_size) { in opj_tgt_init()
188 memset(((char *) p_tree->nodes) + p_tree->nodes_size, 0, in opj_tgt_init()
189 l_node_size - p_tree->nodes_size); in opj_tgt_init()
190 p_tree->nodes_size = l_node_size; in opj_tgt_init()
Dtgt.h72 OPJ_UINT32 nodes_size; /* maximum size taken by nodes */ member
/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dresources.cc57 DCHECK_LT(node_id, tree_ensemble_->trees(tree_id).nodes_size()); in next_node()
84 DCHECK_LT(node_id, tree_ensemble_->trees(tree_id).nodes_size()); in node_value()
120 DCHECK_LT(node_id, tree_ensemble_->trees(tree_id).nodes_size()); in set_node_value()
156 return tree_ensemble_->trees(tree_id).nodes_size(); in GetNumNodes()
166 DCHECK_LT(node_id, tree_ensemble_->trees(tree_id).nodes_size()); in is_leaf()
244 (tree_ensemble_->trees(n_trees - 1).nodes_size() == 1))) { in UpdateGrowingMetadata()
274 *left_node_id = tree->nodes_size(); in AddBucketizedSplitNode()
308 int32 num_nodes = tree->nodes_size(); in PostPruneTree()
/external/tensorflow/tensorflow/lite/
Doptional_debug_tools.cc82 interpreter->tensors_size(), interpreter->nodes_size()); in PrintInterpreterState()
98 for (size_t node_index = 0; node_index < interpreter->nodes_size(); in PrintInterpreterState()
Dinterpreter.h238 size_t nodes_size() const { return primary_subgraph().nodes_size(); } in nodes_size() function
Dmodel_test.cc131 ASSERT_EQ(interpreter->nodes_size(), 2); in TEST()
Dnnapi_delegate.cc220 for (size_t i = 0; i < subgraph->nodes_size(); i++) { in AddOpsAndParams()
722 for (size_t i = 0; i < subgraph->nodes_size(); i++) { in BuildGraph()
Dinterpreter_test.cc105 ASSERT_EQ(interpreter.nodes_size(), 0); in TEST()
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/common/partitioners/
Dexample_partitioner.cc37 if (TF_PREDICT_TRUE(tree_config.nodes_size() > 0)) { in UpdatePartitions()
70 if (TF_PREDICT_TRUE(tree_config.nodes_size() > 0)) { in PartitionExamples()
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/
Ddecision-tree-resource.cc57 int32 newid = tree->nodes_size(); in SplitNode()
85 if (tree->nodes_size() == 0) { in MaybeInitialize()
Ddecision-tree-resource.h37 decision_tree_->decision_tree().nodes_size(), "]"); in DebugString()
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/testutil/
Drandom_tree_gen.cc133 const int left_branch_size = left_branch->nodes_size(); in Combine()
134 CHECK_EQ(1, root->nodes_size()); in Combine()
/external/tensorflow/tensorflow/core/kernels/tensor_forest/
Dresources.cc27 return decision_tree_->nodes_size(); in get_size()
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.h164 size_t nodes_size() const { return nodes_and_registration_.size(); } in nodes_size() function
190 if (node_index < 0 || static_cast<size_t>(node_index) >= nodes_size()) in node_and_registration()
Dsubgraph.cc803 auto nodes_size = nodes_and_registration_.size(); in GetNodeAndRegistration() local
804 TF_LITE_ENSURE(context_, static_cast<size_t>(node_index) < nodes_size); in GetNodeAndRegistration()
/external/tensorflow/tensorflow/lite/experimental/writer/
Dwriter_lib.cc54 operator_to_opcode.resize(interpreter_->nodes_size(), -1); in ExportOperators()
128 for (int op_index = 0; op_index < interpreter_->nodes_size(); ++op_index) { in ExportTensors()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
Dnode_info_delegate_test.cc130 EXPECT_EQ(interpreter->nodes_size(), node_to_opinfo.size()); in TEST()
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/trees/
Ddecision_tree.cc28 if (TF_PREDICT_FALSE(config.nodes_size() <= sub_root_id)) { in Traverse()
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
Dstats_ops.cc388 tree_resource->decision_tree().decision_tree().nodes_size() < in Compute()
480 int num_nodes = tree_resource->decision_tree().decision_tree().nodes_size(); in Compute()
Dmodel_ops.cc143 decision_tree_resource->decision_tree().decision_tree().nodes_size(); in Compute()
/external/tensorflow/tensorflow/lite/examples/label_image/
Dlabel_image.cc120 LOG(INFO) << "nodes size: " << interpreter->nodes_size() << "\n"; in RunInference()
/external/tensorflow/tensorflow/contrib/lite/
Dnnapi_delegate.cc251 for (size_t i = 0; i < interpreter->nodes_size(); i++) { in AddOpsAndParams()
733 for (size_t i = 0; i < interpreter->nodes_size(); i++) { in BuildGraph()