saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

LabelEncoder kernel creation improvement (#16516)

Browse source
This commit is contained in:
Aditya Goel 2023-07-05 16:09:20 +02:00 committed by GitHub
parent e2526714e2
commit 9799d43c36
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
onnxruntime/core/providers/cpu/ml/label_encoder.h
View file

@ -63,9 +63,9 @@ class LabelEncoder_2 final : public OpKernel {
"(name: ", info.node().Name(), ") must have the same length. ",
"However, the number of key is ", num_keys, " and the number of ",
"values is ", num_values, ".");
_map.reserve(num_keys);
for (size_t i = 0; i < num_keys; ++i)
_map[keys[i]] = values[i];
_map.emplace(keys[i], values[i]);
}
Status Compute(OpKernelContext* context) const override {
@ -98,7 +98,7 @@ class LabelEncoder_2 final : public OpKernel {
// A collection of key-value pairs. Each (a_key, a_value) pair
// means that the "a_key" in the input would be mapped to "a_value".
// If _map doesn't contain "a_key", we use _default_value as its output.
std::unordered_map<TKey, TValue> _map;
InlinedHashMap<TKey, TValue> _map;
TValue _default_value;
// ONNX attribute name to load keys.
std::string _key_field_name;