Searched refs:total_num_features (Results 1 – 3 of 3) sorted by relevance
302 int64 total_num_features = 0; in GetSparseTensorShapes() local307 total_num_features += num_elements; in GetSparseTensorShapes()310 output_shapes->indices_shape.AddDim(total_num_features); in GetSparseTensorShapes()312 output_shapes->values_shape.AddDim(total_num_features); in GetSparseTensorShapes()
1082 size_t total_num_features = 0; in FastParseExample() local1087 total_num_features += end_indices.back(); in FastParseExample()1096 indices_shape.AddDim(total_num_features); in FastParseExample()1102 values_shape.AddDim(total_num_features); in FastParseExample()
796 int64 total_num_features = 0; in Compute() local801 total_num_features += num_elements; in Compute()805 TensorShape indices_shape({total_num_features, 2}); in Compute()806 TensorShape values_shape({total_num_features}); in Compute()