Home
last modified time | relevance | path

Searched refs:cached_node_ids (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/
Dprediction_ops_test.py43 cached_node_ids = [0, 0]
54 cached_node_ids=cached_node_ids,
62 self.assertAllClose(cached_node_ids, new_node_ids)
113 cached_node_ids = [0, 0]
121 cached_node_ids=cached_node_ids,
183 cached_node_ids = [1, 2]
194 cached_node_ids=cached_node_ids,
202 self.assertAllClose(cached_node_ids, new_node_ids)
294 cached_node_ids = [1, 0]
304 cached_node_ids=cached_node_ids,
[all …]
/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dprediction_ops.cc83 const auto cached_node_ids = cached_node_ids_t->vec<int32>(); in Compute() local
115 &cached_node_ids, &output_partial_logits, in Compute()
119 int32 node_id = cached_node_ids(i); in Compute()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_BoostedTreesTrainingPredict.pbtxt12 name: "cached_node_ids"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt501 …argspec: "args=[\'tree_ensemble_handle\', \'cached_tree_ids\', \'cached_node_ids\', \'bucketized_f…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt501 …argspec: "args=[\'tree_ensemble_handle\', \'cached_tree_ids\', \'cached_node_ids\', \'bucketized_f…
/external/tensorflow/tensorflow/go/op/
Dwrappers.go3474 …e *Scope, tree_ensemble_handle tf.Output, cached_tree_ids tf.Output, cached_node_ids tf.Output, bu…
3482 tree_ensemble_handle, cached_tree_ids, cached_node_ids, tf.OutputList(bucketized_features),
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt4833 name: "cached_node_ids"
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v1.pbtxt12047 name: "cached_node_ids"
Dops_history.v2.pbtxt12047 name: "cached_node_ids"