Home
last modified time | relevance | path

Searched refs:node_ids (Results 1 – 25 of 44) sorted by relevance

12

/external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/
Dstats_ops_test.py136 (node_ids, gains, _, _, left_node_contribs, right_node_contribs,
149 self.assertAllEqual([1, 2], node_ids)
159 (node_ids, gains, feature_dimensions, thresholds, left_node_contribs,
171 self.assertAllEqual([1, 2], node_ids)
237 (node_ids, gains, feature_ids, feature_dimensions, thresholds,
252 self.assertAllEqual([1, 2], node_ids)
274 (node_ids, gains, feature_ids, feature_dimensions, thresholds,
288 self.assertAllEqual([1, 2], node_ids)
309 (node_ids, gains, feature_ids, feature_dimensions, thresholds,
323 self.assertAllEqual([1, 2], node_ids)
[all …]
Dtraining_ops_test.py77 node_ids=[feature1_nodes, feature2_nodes, feature3_nodes],
185 node_ids=[group1_nodes, group2_nodes],
294 node_ids=[group1_nodes, group2_nodes],
407 node_ids=[group1_nodes, group2_nodes],
620 node_ids=[feature1_nodes, feature2_nodes, feature3_nodes],
823 node_ids=[group1_nodes, group2_nodes, group3_nodes],
1026 node_ids=[group1_nodes, group2_nodes, group3_nodes],
1241 node_ids=[group1_nodes, group2_nodes, group3_nodes],
1462 node_ids=[feature1_nodes],
1615 node_ids=[group1_nodes],
[all …]
/external/google-fruit/include/fruit/impl/data_structures/
Dsemistatic_graph.templates.h87 …HashSetWithArenaAllocator<NodeId> node_ids = createHashSetWithArenaAllocator<NodeId>(last - first,… in SemistaticGraph() local
89 node_ids.insert(i->getId()); in SemistaticGraph()
92 node_ids.insert(*j); in SemistaticGraph()
100 indexing_iterator<itr_t, sizeof(NodeData)>{node_ids.begin(), 0}, in SemistaticGraph()
101 … indexing_iterator<itr_t, sizeof(NodeData)>{node_ids.end(), node_ids.size() * sizeof(NodeData)}, in SemistaticGraph()
102 node_ids.size(), in SemistaticGraph()
105 first_unused_index = node_ids.size(); in SemistaticGraph()
154 node_ids_t node_ids = node_ids_t(ArenaAllocator<node_ids_elem_t>(memory_pool)); in SemistaticGraph() local
157 node_ids.push_back(std::make_pair(i->getId(), InternalNodeId())); in SemistaticGraph()
162 node_ids.push_back(std::make_pair(*j, InternalNodeId())); in SemistaticGraph()
[all …]
/external/tensorflow/tensorflow/python/training/tracking/
Dgraph_view.py89 def _serialize_slot_variables(trackable_objects, node_ids, object_names): argument
119 if slot_variable in node_ids:
130 node_ids[slot_variable] = slot_variable_node_id
232 self, trackable_objects, object_graph_proto, node_ids, object_names, argument
247 assert node_ids[trackable] == checkpoint_id
341 node_ids, argument
349 assert node_ids[trackable] == checkpoint_id
354 child_proto.node_id = node_ids[child.ref]
365 node_ids = object_identity.ObjectIdentityDictionary()
367 node_ids[node] = node_id
[all …]
/external/autotest/client/common_lib/cros/cfm/
Dcras_node_collector_unittest.py49 node_ids = set([node.node_id for node in nodes])
52 self.assertEquals(node_ids, set(
90 node_ids = set([node.node_id for node in nodes])
93 self.assertEquals(node_ids, set(
/external/libchrome/base/containers/
Dlinked_list_unittest.cc50 int num_nodes, const int* node_ids, bool forward) { in ExpectListContentsForDirection() argument
57 EXPECT_EQ(node_ids[index_of_id], node->value()->id()); in ExpectListContentsForDirection()
65 const int* node_ids) { in ExpectListContents() argument
68 ExpectListContentsForDirection(list, num_nodes, node_ids, true); in ExpectListContents()
72 ExpectListContentsForDirection(list, num_nodes, node_ids, false); in ExpectListContents()
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DBoostedTreesUpdateEnsembleV2.pbtxt17 name: "node_ids"
81 name: "node_ids"
153 name: "node_ids"
DBoostedTreesCalculateBestFeatureSplit.pbtxt28 name: "node_ids"
101 name: "node_ids"
DBoostedTreesAggregateStats.pbtxt4 name: "node_ids"
DBoostedTreesTrainingPredict.pbtxt29 name: "node_ids"
DBoostedTreesMakeStatsSummary.pbtxt4 name: "node_ids"
DBoostedTreesSparseAggregateStats.pbtxt4 name: "node_ids"
DBoostedTreesUpdateEnsemble.pbtxt12 name: "node_ids"
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_transformer.h95 const std::string& node_ids,
99 const std::string& node_ids,
/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dtraining_ops.cc179 const auto& node_ids = node_ids_list[feature_idx].vec<int32>(); in FindBestSplitsPerNode() local
187 for (size_t candidate_idx = 0; candidate_idx < node_ids.size(); in FindBestSplitsPerNode()
190 const auto& node_id = node_ids(candidate_idx); in FindBestSplitsPerNode()
399 const auto& node_ids = node_ids_list[group_idx].vec<int32>(); in FindBestSplitsPerNode() local
410 for (size_t candidate_idx = 0; candidate_idx < node_ids.size(); in FindBestSplitsPerNode()
413 const auto& node_id = node_ids(candidate_idx); in FindBestSplitsPerNode()
Dstats_ops.cc1255 const auto node_ids = node_ids_t->vec<int32>(); in Compute() local
1284 const int32 node = node_ids(i); in Compute()
1322 const auto node_ids = node_ids_t->vec<int32>(); in Compute() local
1358 const int32 node = node_ids(i); in Compute()
1471 const TTypes<int32>::ConstVec& node_ids) { in AddInstanceStatsToMap() argument
1472 const int32 node_id = node_ids(instance); in AddInstanceStatsToMap()
1495 const TTypes<int32>::ConstVec& node_ids, in AddRangeStats() argument
1509 stats_map, gradients, hessians, node_ids); in AddRangeStats()
1527 const auto node_ids = node_ids_t->vec<int32>(); in Compute() local
1580 const int32 node_id = node_ids(instance); in Compute()
[all …]
/external/tensorflow/tensorflow/python/saved_model/
Dfunction_serialization.py60 def serialize_concrete_function(concrete_function, node_ids, coder): argument
65 bound_inputs.append(node_ids[capture])
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DBoostedTreesAggregateStats.pbtxt4 name: "node_ids"
DBoostedTreesTrainingPredict.pbtxt29 name: "node_ids"
DBoostedTreesMakeStatsSummary.pbtxt4 name: "node_ids"
DBoostedTreesSparseAggregateStats.pbtxt4 name: "node_ids"
DBoostedTreesUpdateEnsemble.pbtxt12 name: "node_ids"
DBoostedTreesCalculateBestFeatureSplit.pbtxt28 name: "node_ids"
DBoostedTreesCalculateBestFeatureSplitV2.pbtxt37 name: "node_ids"
DBoostedTreesSparseCalculateBestFeatureSplit.pbtxt36 name: "node_ids"

12