Searched defs:feature_size (Results 1 – 6 of 6) 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()
92 int feature_size() const { return sub_features_.size(); } in feature_size() function126 int feature_size() const { return features_.size(); } in feature_size() function
447 const int feature_size, const int id) { in write_features_to_file()
488 int64_t feature_size = weight0.size(-1); in mkldnn_reorder_mkldnn_rnn_layer_weight() local
438 int feature_size = vec_mean.size(1); in batch_norm_reduce_statistics_kernel() local820 auto feature_size = input_reshaped.size(2); in batch_norm_backward_reduce_cuda_template() local
114 const int feature_size, const int id, in write_features_to_file()