Searched defs:feature_size (Results 1 – 3 of 3) sorted by relevance
49 Tensor make_dense_default(int feature_size) { in make_dense_default()63 Tensor make_dense_default(int feature_size) { in make_dense_default()77 Tensor make_dense_default(int feature_size) { in make_dense_default()154 static Graph* ParseExample(int batch_size, int num_keys, int feature_size) { in ParseExample()205 static Graph* ParseExampleV2(int batch_size, int num_keys, int feature_size) { in ParseExampleV2()282 static Graph* ParseSingleExample(int num_keys, int feature_size) { in ParseSingleExample()
91 int feature_size() const { return sub_features_.size(); } in feature_size() function125 int feature_size() const { return features_.size(); } in feature_size() function
22 FeatureProvider::FeatureProvider(int feature_size, int8_t* feature_data) in FeatureProvider()