Home
last modified time | relevance | path

Searched refs:hitShape (Results 1 – 3 of 3) sorted by relevance

/frameworks/ml/nn/common/
DOperationsUtils.cpp481 const Shape& valueShape, Shape* outputShape, Shape* hitShape) { in hashtableLookupPrepare() argument
495 hitShape->type = OperandType::TENSOR_QUANT8_ASYMM; in hashtableLookupPrepare()
496 hitShape->dimensions = {lookups}; in hashtableLookupPrepare()
497 hitShape->offset = 0; in hashtableLookupPrepare()
498 hitShape->scale = 1.f; in hashtableLookupPrepare()
DCpuExecutor.cpp1020 Shape outputShape, hitShape; in executeOperation() local
1024 &outputShape, &hitShape) && in executeOperation()
1026 setInfoAndAllocateIfNeeded(&hits, hitShape, &result) && lookup.Eval(); in executeOperation()
/frameworks/ml/nn/common/include/
DOperationsUtils.h316 const Shape& valueShape, Shape* outputShape, Shape* hitShape);