• Home
  • Raw
  • Download

Lines Matching refs:scales

40   std::vector<float> scales{{0.1, 9.0, -7.0}};  in TEST()  local
51 DequantizeAdd(scales.data(), embeddings.data(), bytes_per_embedding, in TEST()
69 DequantizeAdd(scales.data(), embeddings.data(), bytes_per_embedding, in TEST()
92 std::vector<float> scales(num_buckets); in TEST() local
94 std::fill(scales.begin(), scales.end(), 1); in TEST()
98 DequantizeAdd(scales.data(), embeddings.data(), bytes_per_embedding, in TEST()
115 std::vector<float> scales(num_buckets, 1.0); in TEST() local
119 DequantizeAdd(scales.data(), embeddings.data(), bytes_per_embedding, in TEST()
135 std::vector<float> scales(num_buckets, 1.0); in TEST() local
136 scales[1] = 9.0; in TEST()
144 DequantizeAdd(scales.data(), embeddings.data(), bytes_per_embedding, in TEST()
149 expected.push_back(1.0 / num_sparse_features * (1 - 4) * scales[bucket_id]); in TEST()
150 expected.push_back(1.0 / num_sparse_features * (2 - 4) * scales[bucket_id]); in TEST()
151 expected.push_back(1.0 / num_sparse_features * (3 - 4) * scales[bucket_id]); in TEST()
152 expected.push_back(1.0 / num_sparse_features * (4 - 4) * scales[bucket_id]); in TEST()
153 expected.push_back(1.0 / num_sparse_features * (5 - 4) * scales[bucket_id]); in TEST()
154 expected.push_back(1.0 / num_sparse_features * (6 - 4) * scales[bucket_id]); in TEST()
155 expected.push_back(1.0 / num_sparse_features * (7 - 4) * scales[bucket_id]); in TEST()
156 expected.push_back(1.0 / num_sparse_features * (0 - 4) * scales[bucket_id]); in TEST()
157 expected.push_back(1.0 / num_sparse_features * (0 - 4) * scales[bucket_id]); in TEST()
158 expected.push_back(1.0 / num_sparse_features * (0 - 4) * scales[bucket_id]); in TEST()