Searched refs:lookups (Results 1 - 7 of 7) sorted by relevance
| /frameworks/ml/nn/runtime/test/specs/V1_0/ |
| H A D | hashtable_lookup_float.mod.py | 17 lookups = 4 variable
29 lookup = Input("lookup", "TENSOR_INT32", "{%d}" % lookups)
32 output = Output("output", "TENSOR_FLOAT32", "{%d, %d}" % (lookups, features))
33 hits = Output("hits", "TENSOR_QUANT8_ASYMM", "{%d}, 1.f, 0" % (lookups))
|
| H A D | hashtable_lookup_quant8.mod.py | 17 lookups = 4 variable
29 lookup = Input("lookup", "TENSOR_INT32", "{%d}" % (lookups))
32 output = Output("output", "TENSOR_QUANT8_ASYMM", "{%d, %d}, 0.5f, 0" % (lookups, features))
33 hits = Output("hits", "TENSOR_QUANT8_ASYMM", "{%d}, 1.f, 0" % (lookups))
|
| H A D | embedding_lookup.mod.py | 17 lookups = 3 variable
29 index = Input("index", "TENSOR_INT32", "{%d}"%lookups)
31 output = Output("output", "TENSOR_FLOAT32", "{%d, %d, %d}" % (lookups, columns, features))
|
| /frameworks/ml/nn/runtime/test/specs/V1_1/ |
| H A D | hashtable_lookup_float_relaxed.mod.py | 17 lookups = 4 variable
29 lookup = Input("lookup", "TENSOR_INT32", "{%d}" % lookups)
32 output = Output("output", "TENSOR_FLOAT32", "{%d, %d}" % (lookups, features))
33 hits = Output("hits", "TENSOR_QUANT8_ASYMM", "{%d}, 1.f, 0" % (lookups))
|
| H A D | embedding_lookup_relaxed.mod.py | 17 lookups = 3 variable
29 index = Input("index", "TENSOR_INT32", "{%d}"%lookups)
31 output = Output("output", "TENSOR_FLOAT32", "{%d, %d, %d}" % (lookups, columns, features))
|
| /frameworks/ml/nn/common/ |
| H A D | OperationsUtils.cpp | 528 const uint32_t lookups = getSizeOfDimension(lookupShape, 0); local
531 outputShape->dimensions = { lookups, columns };
550 const uint32_t lookups = getSizeOfDimension(lookupShape, 0); local
554 outputShape->dimensions = { lookups };
562 hitShape->dimensions = { lookups };
|
| H A D | CpuExecutor.cpp | 1211 const RunTimeOperandInfo &lookups = local
1219 success = embeddingLookupPrepare(values.shape(), lookups.shape(), &outputShape) &&
1224 const RunTimeOperandInfo &lookups = local
1239 success = hashtableLookupPrepare(lookups.shape(), keys.shape(), values.shape(),
|
Completed in 141 milliseconds