Home
last modified time | relevance | path

Searched refs:total_num_features (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/parse_example/
Dexample_proto_fast_parsing.cc26 size_t* total_num_features, size_t* max_num_features) { in CountSparseFeatures() argument
30 *total_num_features += end_indices.back(); in CountSparseFeatures()
Dparse_example.cc392 size_t* total_num_features, size_t* max_num_features) { in CountSparseFeatures() argument
394 *total_num_features += end_indices.back(); in CountSparseFeatures()
483 size_t total_num_features = 0; in FastParseExampleLite() local
485 CountSparseFeatures(sparse_buffers[d], &total_num_features, in FastParseExampleLite()
496 index_shape->data[0] = total_num_features; in FastParseExampleLite()
501 output_shape->data[0] = total_num_features; in FastParseExampleLite()
512 if (total_num_features > 0) { in FastParseExampleLite()
521 if (idx0 < total_num_features) { in FastParseExampleLite()
Dexample_proto_fast_parsing.h425 size_t* total_num_features, size_t* max_num_features);
/external/tensorflow/tensorflow/core/util/
Dexample_proto_helper.cc305 int64 total_num_features = 0; in GetSparseTensorShapes() local
310 total_num_features += num_elements; in GetSparseTensorShapes()
313 output_shapes->indices_shape.AddDim(total_num_features); in GetSparseTensorShapes()
315 output_shapes->values_shape.AddDim(total_num_features); in GetSparseTensorShapes()
Dexample_proto_fast_parsing.cc1076 size_t* total_num_features, size_t* max_num_features) { in CountSparseFeatures() argument
1080 *total_num_features += end_indices.back(); in CountSparseFeatures()
1253 size_t total_num_features = 0; in FastParseExample() local
1255 CountSparseFeatures(sparse_buffers, d, &total_num_features, in FastParseExample()
1259 indices_shape.AddDim(total_num_features); in FastParseExample()
1265 values_shape.AddDim(total_num_features); in FastParseExample()
1306 size_t total_num_features = 0; in FastParseExample() local
1308 CountSparseFeatures(ragged_buffers, d, &total_num_features, in FastParseExample()
1323 values_shape.AddDim(total_num_features); in FastParseExample()
/external/tensorflow/tensorflow/core/kernels/
Dexample_parsing_ops.cc1152 int64 total_num_features = 0; in Compute() local
1157 total_num_features += num_elements; in Compute()
1161 TensorShape indices_shape({total_num_features, 2}); in Compute()
1162 TensorShape values_shape({total_num_features}); in Compute()