Searched refs:max_num_features (Results 1 – 4 of 4) sorted by relevance
303 int64 max_num_features = 0; in GetSparseTensorShapes() local308 max_num_features = std::max(max_num_features, num_elements); in GetSparseTensorShapes()313 output_shapes->max_num_features = max_num_features; in GetSparseTensorShapes()390 shape_t(1) = sparse_tensor_batch_shapes.max_num_features; in BatchExampleProtoToTensors()
1083 size_t max_num_features = 0; in FastParseExample() local1088 max_num_features = std::max(max_num_features, end_indices[0]); in FastParseExample()1091 max_num_features = std::max(max_num_features, example_size); in FastParseExample()1109 shapes_shape_t(1) = max_num_features; in FastParseExample()1163 size_t max_num_features = 0; in FastParseExample() local1167 max_num_features = std::max(max_num_features, end_indices[0]); in FastParseExample()1170 max_num_features = std::max(max_num_features, example_size); in FastParseExample()1175 const size_t max_num_elements = max_num_features / stride_size; in FastParseExample()1177 DCHECK_EQ(max_num_features % config.dense[d].elements_per_stride, 0); in FastParseExample()
88 int max_num_features; member
797 int64 max_num_features = 0; in Compute() local802 max_num_features = std::max(max_num_features, num_elements); in Compute()818 shape_t(1) = max_num_features; in Compute()