/external/libtextclassifier/ |
quantization_test.cc | 47 const int num_sparse_features = 7; local 52 num_sparse_features, quantization_bits, bucket_id, 70 num_sparse_features, quantization_bits, bucket_id, 88 const int num_sparse_features = 7; local 99 num_sparse_features, quantization_bits, bucket_id, dest.data(), 104 1.0 / num_sparse_features * (0 - 1)); 111 const int num_sparse_features = 7; local 120 num_sparse_features, quantization_bits, bucket_id, dest.data(), 124 1.0 / num_sparse_features * (1 - 1)); 131 const int num_sparse_features = 7 local [all...] |
model-executor.cc | 109 const int num_sparse_features = sparse_features.size(); local 110 for (int i = 0; i < num_sparse_features; ++i) { 117 bytes_per_embedding_, num_sparse_features,
|
/external/tensorflow/tensorflow/core/kernels/ |
sdca_ops.cc | 85 context, context->GetAttr("num_sparse_features", &num_sparse_features)); 91 context, num_sparse_features + num_dense_features > 0, 95 static_cast<int64>(num_sparse_features) + 100 static_cast<int64>(num_sparse_features) + 111 int num_sparse_features = 0; member in struct:tensorflow::__anon39911::ComputeOptions 130 examples.Initialize(context, model_weights, options.num_sparse_features,
|
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
quantile_ops.cc | 50 const char* const kNumSparseFeaturesName = "num_sparse_features"; 147 int* num_sparse_features) { 151 context->GetAttr(kNumSparseFeaturesName, num_sparse_features)); 152 if ((*num_dense_features) + (*num_sparse_features) == 0) { 863 int num_sparse_features; local [all...] |