Home
last modified time | relevance | path

Searched refs:feature_size (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dexample_parsing_ops_test.cc43 void operator()(Feature* f, int feature_size) const { in operator ()()
44 for (int i = 0; i < feature_size; ++i) { in operator ()()
48 Tensor make_dense_default(int feature_size) { in make_dense_default() argument
49 return Tensor(dtype, TensorShape({feature_size})); in make_dense_default()
57 void operator()(Feature* f, int feature_size) const { in operator ()()
58 for (int i = 0; i < feature_size; ++i) { in operator ()()
62 Tensor make_dense_default(int feature_size) { in make_dense_default() argument
63 return Tensor(dtype, TensorShape({feature_size})); in make_dense_default()
71 void operator()(Feature* f, int feature_size) const { in operator ()()
72 for (int i = 0; i < feature_size; ++i) { in operator ()()
[all …]
Dexample_parsing_ops.cc742 out_shape.AddDim(fl.feature_size()); in Compute()
750 for (int64 t = 0; t < fl.feature_size(); ++t) { in Compute()
778 feature_list_size = fl.feature_size(); in Compute()
/external/libtextclassifier/lang_id/common/fel/
Dfeature-descriptors.cc45 if (function.feature_size() == 1) { in ToFEL()
48 } else if (function.feature_size() > 1) { in ToFEL()
50 for (int i = 0; i < function.feature_size(); ++i) { in ToFEL()
59 for (int i = 0; i < extractor.feature_size(); ++i) { in ToFEL()
Dfeature-descriptors.h91 int feature_size() const { return sub_features_.size(); } in feature_size() function
93 SAFTM_DCHECK((i >= 0) && (i < feature_size())); in feature()
125 int feature_size() const { return features_.size(); } in feature_size() function
133 SAFTM_DCHECK((i >= 0) && (i < feature_size())); in feature()
Dfeature-extractor.h443 for (int i = 0; i < fd->feature_size(); ++i) { in CreateNested()
627 for (int i = 0; i < descriptor().feature_size(); ++i) { in InitializeFeatureFunctions()
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/testutil/
Drandom_tree_gen.cc189 int feature_size; in GenerateSplit() local
191 feature_size = dense_feature_size_; in GenerateSplit()
194 feature_size = sparse_feature_size_; in GenerateSplit()
204 split->set_feature_column(rng_->Uniform(feature_size)); in GenerateSplit()
/external/tensorflow/tensorflow/lite/experimental/micro/examples/micro_speech/
Dfeature_provider.cc22 FeatureProvider::FeatureProvider(int feature_size, uint8_t* feature_data) in FeatureProvider() argument
23 : feature_size_(feature_size), in FeatureProvider()
Dfeature_provider.h35 FeatureProvider(int feature_size, uint8_t* feature_data);
/external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/
Drnn_cell_test.py406 feature_size = 2
408 num_shifts = (input_size - feature_size) // frequency_skip + 1
415 feature_size=feature_size,
444 feature_size = 2
446 num_shifts = int((input_size - feature_size) / frequency_skip + 1)
451 feature_size=feature_size,
489 feature_size = 2
499 feature_size=feature_size,
540 feature_size = 2
542 num_shifts = int((input_size - feature_size) / frequency_skip + 1)
[all …]
/external/tensorflow/tensorflow/core/example/
Dfeature_util_test.cc363 EXPECT_EQ(1, se.feature_lists().feature_list().at("tag").feature_size()); in TEST()
/external/tensorflow/tensorflow/contrib/rnn/python/ops/
Drnn_cell.py344 feature_size=None, argument
375 self._feature_size = feature_size
523 feature_size=None, argument
584 self._feature_size = feature_size
983 feature_size=None, argument
1030 initializer, num_unit_shards, forget_bias, feature_size, frequency_skip,