Searched refs:batch_features (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/ |
| D | batch_features_test.cc | 34 BatchFeatures batch_features(8); in TEST_F() local
35 EXPECT_DEATH(({ batch_features.Initialize({}, {}, {}, {}, {}, {}, {}); }) in TEST_F()
41 BatchFeatures batch_features(8); in TEST_F() local
46 batch_features.Initialize({dense_vec}, {}, {}, {}, {}, {}, {})); in TEST_F()
50 BatchFeatures batch_features(8); in TEST_F() local
55 batch_features.Initialize({dense_vec}, {}, {}, {}, {}, {}, {})); in TEST_F()
59 BatchFeatures batch_features(1); in TEST_F() local
64 batch_features.Initialize({dense_vec}, {}, {}, {}, {}, {}, {})); in TEST_F()
68 BatchFeatures batch_features(2); in TEST_F() local
74 EXPECT_EQ(expected_error, batch_features.Initialize( in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | prediction_ops.cc | 219 boosted_trees::utils::BatchFeatures batch_features(batch_size); in DoCompute() local
222 batch_features.Initialize( in DoCompute()
306 MultipleAdditiveTrees::Predict(adjusted, trees_to_include, batch_features, in DoCompute()
312 batch_features, worker_threads, output_predictions, in DoCompute()
431 boosted_trees::utils::BatchFeatures batch_features(batch_size); in DoCompute() local
434 batch_features.Initialize( in DoCompute()
455 tree_config, batch_features, worker_threads->NumThreads(), in DoCompute()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/testutil/ |
| D | batch_features_testutil.cc | 29 boosted_trees::utils::BatchFeatures* batch_features) { in RandomlyInitializeBatchFeatures() argument
30 const int64 batch_size = static_cast<int64>(batch_features->batch_size()); in RandomlyInitializeBatchFeatures()
80 TF_EXPECT_OK(batch_features->Initialize( in RandomlyInitializeBatchFeatures()
|
| D | batch_features_testutil.h | 39 boosted_trees::utils::BatchFeatures* batch_features);
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | partial_batch_padding_handler.py | 45 def _find_any_tensor(batch_features): argument
47 x for x in nest.flatten(batch_features) if tensor_util.is_tensor(x)
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/ |
| D | BUILD | 23 "utils/batch_features.cc",
31 "utils/batch_features.h",
|
| /external/tensorflow/tensorflow/contrib/cmake/ |
| D | tf_core_kernels.cmake | 51 "${tensorflow_source_dir}/tensorflow/contrib/boosted_trees/lib/utils/batch_features.cc"
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_estimator.py | 3710 def _find_any_tensor(batch_features): argument
3712 x for x in nest.flatten(batch_features) if isinstance(x, ops.Tensor)
|