HomeSort by relevance Sort by last modified time
    Searched refs:batch_features (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/
batch_features_test.cc 16 #include "tensorflow/contrib/boosted_trees/lib/utils/batch_features.h"
34 BatchFeatures batch_features(8);
35 EXPECT_DEATH(({ batch_features.Initialize({}, {}, {}, {}, {}, {}, {}); })
41 BatchFeatures batch_features(8);
46 batch_features.Initialize({dense_vec}, {}, {}, {}, {}, {}, {}));
50 BatchFeatures batch_features(8);
55 batch_features.Initialize({dense_vec}, {}, {}, {}, {}, {}, {}));
59 BatchFeatures batch_features(1);
64 batch_features.Initialize({dense_vec}, {}, {}, {}, {}, {}, {}));
68 BatchFeatures batch_features(2)
    [all...]
  /external/tensorflow/tensorflow/contrib/boosted_trees/lib/testutil/
batch_features_testutil.h 18 #include "tensorflow/contrib/boosted_trees/lib/utils/batch_features.h"
26 // This method calls Initialize on the given 'batch_features', which will be
29 // because they must outlive 'batch_features'.
39 boosted_trees::utils::BatchFeatures* batch_features);
batch_features_testutil.cc 29 boosted_trees::utils::BatchFeatures* batch_features) {
30 const int64 batch_size = static_cast<int64>(batch_features->batch_size());
80 TF_EXPECT_OK(batch_features->Initialize(
  /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
prediction_ops.cc 21 #include "tensorflow/contrib/boosted_trees/lib/utils/batch_features.h"
219 boosted_trees::utils::BatchFeatures batch_features(batch_size);
222 batch_features.Initialize(
306 MultipleAdditiveTrees::Predict(adjusted, trees_to_include, batch_features,
312 batch_features, worker_threads, output_predictions,
431 boosted_trees::utils::BatchFeatures batch_features(batch_size);
434 batch_features.Initialize(
455 tree_config, batch_features, worker_threads->NumThreads(),
  /external/tensorflow/tensorflow/python/keras/engine/
partial_batch_padding_handler.py 45 def _find_any_tensor(batch_features):
47 x for x in nest.flatten(batch_features) if tensor_util.is_tensor(x)

Completed in 381 milliseconds